diff --git a/tests/benchmark/milvus_benchmark/env/local.py b/tests/benchmark/milvus_benchmark/env/local.py
index 0d5d4cffc3..d6126c5686 100644
--- a/tests/benchmark/milvus_benchmark/env/local.py
+++ b/tests/benchmark/milvus_benchmark/env/local.py
@@ -16,6 +16,7 @@ class LocalEnv(BaseEnv):
         res = True
         try:
             self.set_hostname(hostname)
+            self.set_port(port)
         except Exception as e:
             logger.error(str(e))
             res = False