diff --git a/client/proxy/proxy.go b/client/proxy/proxy.go
index 0c43740f..f5e1b603 100644
--- a/client/proxy/proxy.go
+++ b/client/proxy/proxy.go
@@ -521,7 +521,7 @@ func HandleTcpWorkConnection(localInfo *config.LocalSvrConf, proxyPlugin plugin.
 				DestinationPort:    m.DstPort,
 			}
 
-			if h.SourceAddress.To16() == nil {
+			if strings.Contains(m.SrcAddr, ".") {
 				h.TransportProtocol = pp.TCPv4
 			} else {
 				h.TransportProtocol = pp.TCPv6