diff --git a/pymilvus/orm/connections.py b/pymilvus/orm/connections.py
index a90c778db..9fc60f491 100644
--- a/pymilvus/orm/connections.py
+++ b/pymilvus/orm/connections.py
@@ -319,7 +319,7 @@ def with_config(config: Tuple) -> bool:
         uri = kwargs.get("uri")
         if uri is not None:
             server_type = get_server_type(uri)
-            if server_type == ZILLIZ and ":" not in token:
+            if server_type == ZILLIZ and ":" not in uri:
                 kwargs["uri"] = uri + ":" + str(VIRTUAL_PORT)
 
         config = (