diff --git a/proxy/socks/server_config_json_test.go b/proxy/socks/server_config_json_test.go
index 74be8a24..086c83b5 100644
--- a/proxy/socks/server_config_json_test.go
+++ b/proxy/socks/server_config_json_test.go
@@ -6,7 +6,7 @@ import (
 	"testing"
 
 	"v2ray.com/core/proxy/registry"
-	"v2ray.com/core/proxy/socks"
+	. "v2ray.com/core/proxy/socks"
 	"v2ray.com/core/testing/assert"
 )
 
@@ -17,5 +17,5 @@ func TestDefaultIPAddress(t *testing.T) {
     "auth": "noauth"
   }`))
 	assert.Error(err).IsNil()
-	assert.Address(socksConfig.(*socks.Config).Address).EqualsString("127.0.0.1")
+	assert.Address(socksConfig.(*ServerConfig).Address.AsAddress()).EqualsString("127.0.0.1")
 }