diff --git a/agent/pkg/agent/config.go b/agent/pkg/agent/config.go
index e727905..d0ad293 100644
--- a/agent/pkg/agent/config.go
+++ b/agent/pkg/agent/config.go
@@ -1,6 +1,7 @@
 package agent
 
 import (
+	"context"
 	"crypto/tls"
 	"crypto/x509"
 	"encoding/json"
@@ -18,7 +19,6 @@ import (
 	"github.com/yusing/go-proxy/internal/net/types"
 	"github.com/yusing/go-proxy/internal/task"
 	"github.com/yusing/go-proxy/pkg"
-	"golang.org/x/net/context"
 )
 
 type AgentConfig struct {
diff --git a/agent/pkg/agent/requests.go b/agent/pkg/agent/requests.go
index a50e9f0..b9b810a 100644
--- a/agent/pkg/agent/requests.go
+++ b/agent/pkg/agent/requests.go
@@ -1,11 +1,11 @@
 package agent
 
 import (
+	"context"
 	"io"
 	"net/http"
 
 	"github.com/coder/websocket"
-	"golang.org/x/net/context"
 )
 
 func (cfg *AgentConfig) Do(ctx context.Context, method, endpoint string, body io.Reader) (*http.Response, error) {