diff --git a/tests/benchmark/milvus_benchmark/update.py b/tests/benchmark/milvus_benchmark/update.py
index 1cabc99048..c6969efe09 100644
--- a/tests/benchmark/milvus_benchmark/update.py
+++ b/tests/benchmark/milvus_benchmark/update.py
@@ -163,11 +163,11 @@ def update_values(src_values_file, deploy_params_file):
 
     cluster = False
     # values_dict["service"]["type"] = "ClusterIP"
-    if deploy_mode == config.CLUSTER_DEPLOY_MODE:
+    if deploy_mode in [config.CLUSTER_DEPLOY_MODE, config.CLUSTER_3RD_DEPLOY_MODE]:
         cluster = True
         # values_dict["cluster"]["enabled"] = True
-    elif deploy_mode == config.CLUSTER_3RD_DEPLOY_MODE:
-        cluster = True
+    # elif deploy_mode == config.CLUSTER_3RD_DEPLOY_MODE:
+    #     cluster = True
     elif deploy_mode == config.SINGLE_DEPLOY_MODE:
         values_dict["cluster"]["enabled"] = False
         values_dict["etcd"]["replicaCount"] = 1