diff --git a/internal/skysocks/server_test.go b/internal/skysocks/server_test.go
index 123ed6f103..094f533779 100644
--- a/internal/skysocks/server_test.go
+++ b/internal/skysocks/server_test.go
@@ -11,7 +11,6 @@ import (
 	"time"
 
 	"github.com/SkycoinProject/skycoin/src/util/logging"
-	"github.com/hashicorp/yamux"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	"golang.org/x/net/nettest"
@@ -50,13 +49,9 @@ func TestProxy(t *testing.T) {
 	conn, err := net.Dial("tcp", l.Addr().String())
 	require.NoError(t, err)
 
-	session, err := yamux.Client(conn, nil)
+	client, err := NewClient(conn)
 	require.NoError(t, err)
 
-	client := &Client{
-		session: session,
-	}
-
 	errChan2 := make(chan error)
 	go func() {
 		errChan2 <- client.ListenAndServe(":10080")