diff --git a/docs/auth/silly/access_test.go b/docs/auth/silly/access_test.go
index 8b5ecb80..ff2155b1 100644
--- a/docs/auth/silly/access_test.go
+++ b/docs/auth/silly/access_test.go
@@ -5,8 +5,8 @@ import (
 	"net/http/httptest"
 	"testing"
 
+	"github.com/docker/distribution/context"
 	"github.com/docker/distribution/registry/auth"
-	"golang.org/x/net/context"
 )
 
 func TestSillyAccessController(t *testing.T) {
diff --git a/docs/auth/token/token_test.go b/docs/auth/token/token_test.go
index 9d84d4ef..119aa738 100644
--- a/docs/auth/token/token_test.go
+++ b/docs/auth/token/token_test.go
@@ -15,9 +15,9 @@ import (
 	"testing"
 	"time"
 
+	"github.com/docker/distribution/context"
 	"github.com/docker/distribution/registry/auth"
 	"github.com/docker/libtrust"
-	"golang.org/x/net/context"
 )
 
 func makeRootKeys(numKeys int) ([]libtrust.PrivateKey, error) {
diff --git a/docs/handlers/api_test.go b/docs/handlers/api_test.go
index e351cb95..a975bd33 100644
--- a/docs/handlers/api_test.go
+++ b/docs/handlers/api_test.go
@@ -19,6 +19,7 @@ import (
 	"testing"
 
 	"github.com/docker/distribution/configuration"
+	"github.com/docker/distribution/context"
 	"github.com/docker/distribution/digest"
 	"github.com/docker/distribution/manifest"
 	"github.com/docker/distribution/registry/api/errcode"
@@ -27,7 +28,6 @@ import (
 	"github.com/docker/distribution/testutil"
 	"github.com/docker/libtrust"
 	"github.com/gorilla/handlers"
-	"golang.org/x/net/context"
 )
 
 var headerConfig = http.Header{
diff --git a/docs/handlers/app_test.go b/docs/handlers/app_test.go
index fbb0b188..0038a97d 100644
--- a/docs/handlers/app_test.go
+++ b/docs/handlers/app_test.go
@@ -9,6 +9,7 @@ import (
 	"testing"
 
 	"github.com/docker/distribution/configuration"
+	"github.com/docker/distribution/context"
 	"github.com/docker/distribution/registry/api/errcode"
 	"github.com/docker/distribution/registry/api/v2"
 	"github.com/docker/distribution/registry/auth"
@@ -16,7 +17,6 @@ import (
 	"github.com/docker/distribution/registry/storage"
 	memorycache "github.com/docker/distribution/registry/storage/cache/memory"
 	"github.com/docker/distribution/registry/storage/driver/inmemory"
-	"golang.org/x/net/context"
 )
 
 // TestAppDispatcher builds an application with a test dispatcher and ensures
diff --git a/docs/handlers/health_test.go b/docs/handlers/health_test.go
index ce5860a8..38ea9b2f 100644
--- a/docs/handlers/health_test.go
+++ b/docs/handlers/health_test.go
@@ -10,8 +10,8 @@ import (
 	"time"
 
 	"github.com/docker/distribution/configuration"
+	"github.com/docker/distribution/context"
 	"github.com/docker/distribution/health"
-	"golang.org/x/net/context"
 )
 
 func TestFileHealthCheck(t *testing.T) {