diff --git a/gomock/controller.go b/gomock/controller.go
index fa6674c2..6de1f25f 100644
--- a/gomock/controller.go
+++ b/gomock/controller.go
@@ -56,8 +56,8 @@
 package gomock
 
 import (
-	"context"
 	"fmt"
+	"golang.org/x/net/context"
 	"reflect"
 	"runtime"
 	"sync"