diff --git a/tests/benchmark/milvus_benchmark/client.py b/tests/benchmark/milvus_benchmark/client.py
index d7a8f273a2..64f4fa35fa 100644
--- a/tests/benchmark/milvus_benchmark/client.py
+++ b/tests/benchmark/milvus_benchmark/client.py
@@ -418,20 +418,20 @@ class MilvusClient(object):
         logger.debug("Row count: %d in collection: <%s>" % (row_count, collection_name))
         return row_count
 
-    def drop(self, timeout=120, collection_name=None):
+    def drop(self, loop_cunt=120, collection_name=None):
         """
         drop steps:
         1.drop collection
         2.check collection exist
         3.Set timeout to exit
         """
-        timeout = int(timeout)
+        loop_cunt = int(loop_cunt)
         if collection_name is None:
             collection_name = self._collection_name
         logger.info("Start delete collection: %s" % collection_name)
         self._milvus.drop_collection(collection_name)
         i = 0
-        while i < timeout:
+        while i < loop_cunt:
             try:
                 res = self._milvus.has_collection(collection_name)
                 if res:
@@ -443,7 +443,7 @@ class MilvusClient(object):
             except Exception as e:
                 logger.warning("Collection count failed: {}".format(str(e)))
                 break
-        if i >= timeout:
+        if i >= loop_cunt:
             logger.error("Delete collection timeout")
 
     def get_stats(self):