diff --git a/common/mux/client_test.go b/common/mux/client_test.go
index 101ab301c..898ffcc2d 100644
--- a/common/mux/client_test.go
+++ b/common/mux/client_test.go
@@ -9,8 +9,8 @@ import (
 	"v2ray.com/core/common"
 	"v2ray.com/core/common/errors"
 	"v2ray.com/core/common/mux"
-	"v2ray.com/core/common/vio"
 	"v2ray.com/core/testing/mocks"
+	"v2ray.com/core/transport"
 	"v2ray.com/core/transport/pipe"
 )