diff --git a/gobreaker.go b/gobreaker.go
index 67fa2c3..516d45b 100644
--- a/gobreaker.go
+++ b/gobreaker.go
@@ -4,7 +4,7 @@ import (
 	"github.com/micro/go-micro/client"
 	"github.com/sony/gobreaker"
 
-	"golang.org/x/net/context"
+	"context"
 )
 
 type clientWrapper struct {
diff --git a/gobreaker_test.go b/gobreaker_test.go
index 3de0982..e878504 100644
--- a/gobreaker_test.go
+++ b/gobreaker_test.go
@@ -8,7 +8,7 @@ import (
 	"github.com/micro/go-micro/selector"
 	"github.com/sony/gobreaker"
 
-	"golang.org/x/net/context"
+	"context"
 )
 
 func TestBreaker(t *testing.T) {