diff --git a/google/appengine_gen1.go b/google/appengine_gen1.go
index 1ca0517..83dacac 100644
--- a/google/appengine_gen1.go
+++ b/google/appengine_gen1.go
@@ -9,11 +9,11 @@
 package google
 
 import (
+	"context"
 	"sort"
 	"strings"
 	"sync"
 
-	"golang.org/x/net/context"
 	"golang.org/x/oauth2"
 	"google.golang.org/appengine"
 )
diff --git a/google/appengine_gen2_flex.go b/google/appengine_gen2_flex.go
index 819aa68..04c2c22 100644
--- a/google/appengine_gen2_flex.go
+++ b/google/appengine_gen2_flex.go
@@ -9,10 +9,10 @@
 package google
 
 import (
+	"context"
 	"log"
 	"sync"
 
-	"golang.org/x/net/context"
 	"golang.org/x/oauth2"
 )