diff --git a/proxy/ws/ws.go b/proxy/ws/ws.go
index 7d13a1d..4d54d92 100644
--- a/proxy/ws/ws.go
+++ b/proxy/ws/ws.go
@@ -35,9 +35,7 @@ func NewWS(s string, d proxy.Dialer, p proxy.Proxy) (*WS, error) {
 	}
 
 	addr := u.Host
-
-	// TODO:
-	if addr == "" {
+	if addr == "" && d != nil {
 		addr = d.Addr()
 	}