diff --git a/.travis.yml b/.travis.yml
index d6f85ad..47a6a46 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,6 +1,6 @@
 language: go
 go:
-  - 1.3.3
+  - 1.7
   - 1.x
   - tip
 before_install:
diff --git a/context.go b/context.go
index 5d15709..cf29de9 100644
--- a/context.go
+++ b/context.go
@@ -3,7 +3,7 @@ package backoff
 import (
 	"time"
 
-	"golang.org/x/net/context"
+	"context"
 )
 
 // BackOffContext is a backoff policy that stops retrying after the context
diff --git a/context_test.go b/context_test.go
index 993fa61..33002c9 100644
--- a/context_test.go
+++ b/context_test.go
@@ -4,7 +4,7 @@ import (
 	"testing"
 	"time"
 
-	"golang.org/x/net/context"
+	"context"
 )
 
 func TestContext(t *testing.T) {
diff --git a/example_test.go b/example_test.go
index d97a8db..9b68d88 100644
--- a/example_test.go
+++ b/example_test.go
@@ -3,7 +3,7 @@ package backoff
 import (
 	"log"
 
-	"golang.org/x/net/context"
+	"context"
 )
 
 func ExampleRetry() {
diff --git a/retry_test.go b/retry_test.go
index 5af2888..b3c38d3 100644
--- a/retry_test.go
+++ b/retry_test.go
@@ -7,7 +7,7 @@ import (
 	"testing"
 	"time"
 
-	"golang.org/x/net/context"
+	"context"
 )
 
 func TestRetry(t *testing.T) {
diff --git a/ticker_test.go b/ticker_test.go
index 6aae254..8f64d7c 100644
--- a/ticker_test.go
+++ b/ticker_test.go
@@ -7,7 +7,7 @@ import (
 	"testing"
 	"time"
 
-	"golang.org/x/net/context"
+	"context"
 )
 
 func TestTicker(t *testing.T) {