diff --git a/tests/benchmark/milvus_benchmark/client.py b/tests/benchmark/milvus_benchmark/client.py
index 7ba40949497292f69a4bb32b5edb0c22af065388..dadfd5c73efa017f0397c9f776b239ef5019a9d9 100644
--- a/tests/benchmark/milvus_benchmark/client.py
+++ b/tests/benchmark/milvus_benchmark/client.py
@@ -472,7 +472,7 @@ class MilvusClient(object):
         logger.debug("[scene_test] Start insert : %s" % collection_name)
         self.insert(entities, collection_name=collection_name)
         logger.debug("[scene_test] Start flush : %s" % collection_name)
-        self.flush()
+        self.flush(collection_name=collection_name)
 
         logger.debug("[scene_test] Start create index : %s" % collection_name)
         self.create_index(field_name='float_vector', index_type="ivf_sq8", metric_type='l2',