diff --git a/rate/rate.go b/rate/rate.go
index 938feaf..f657292 100644
--- a/rate/rate.go
+++ b/rate/rate.go
@@ -6,12 +6,11 @@
 package rate
 
 import (
+	"context"
 	"fmt"
 	"math"
 	"sync"
 	"time"
-
-	"golang.org/x/net/context"
 )
 
 // Limit defines the maximum frequency of some events.
diff --git a/rate/rate_test.go b/rate/rate_test.go
index cf45d92..9600efa 100644
--- a/rate/rate_test.go
+++ b/rate/rate_test.go
@@ -5,14 +5,13 @@
 package rate
 
 import (
+	"context"
 	"math"
 	"runtime"
 	"sync"
 	"sync/atomic"
 	"testing"
 	"time"
-
-	"golang.org/x/net/context"
 )
 
 func TestLimit(t *testing.T) {