diff --git a/sftp/sftp.go b/sftp/sftp.go
index a7a696b44..3f65aea97 100644
--- a/sftp/sftp.go
+++ b/sftp/sftp.go
@@ -5,7 +5,6 @@
 package sftp
 
 import (
-	"context"
 	"io"
 	"io/ioutil"
 	"os"
@@ -20,6 +19,7 @@ import (
 	"github.com/pkg/sftp"
 	"github.com/xanzy/ssh-agent"
 	"golang.org/x/crypto/ssh"
+	"golang.org/x/net/context"
 	"golang.org/x/time/rate"
 )