diff --git a/transport/internet/websocket/connection.go b/transport/internet/websocket/connection.go
index 3f08cf24..f9e334fc 100644
--- a/transport/internet/websocket/connection.go
+++ b/transport/internet/websocket/connection.go
@@ -5,7 +5,8 @@ import (
 	"net"
 	"time"
 
-	"github.com/gorilla/websocket"
+	"websocket"
+
 	"v2ray.com/core/common/buf"
 	"v2ray.com/core/common/errors"
 )
diff --git a/transport/internet/websocket/dialer.go b/transport/internet/websocket/dialer.go
index 040c9879..f0f5ef2a 100644
--- a/transport/internet/websocket/dialer.go
+++ b/transport/internet/websocket/dialer.go
@@ -4,7 +4,8 @@ import (
 	"context"
 	"time"
 
-	"github.com/gorilla/websocket"
+	"websocket"
+
 	"v2ray.com/core/common"
 	"v2ray.com/core/common/net"
 	"v2ray.com/core/transport/internet"
diff --git a/transport/internet/websocket/hub.go b/transport/internet/websocket/hub.go
index f0a39fc2..5a2591fd 100644
--- a/transport/internet/websocket/hub.go
+++ b/transport/internet/websocket/hub.go
@@ -8,7 +8,8 @@ import (
 	"sync"
 	"time"
 
-	"github.com/gorilla/websocket"
+	"websocket"
+
 	"v2ray.com/core/common"
 	"v2ray.com/core/common/net"
 	http_proto "v2ray.com/core/common/protocol/http"