diff --git a/tests/benchmark/milvus_benchmark/runners/base.py b/tests/benchmark/milvus_benchmark/runners/base.py
index ccd95d25eb..589ad4d10c 100644
--- a/tests/benchmark/milvus_benchmark/runners/base.py
+++ b/tests/benchmark/milvus_benchmark/runners/base.py
@@ -92,6 +92,7 @@ class BaseRunner(object):
 
     # TODO: need to improve
     def insert(self, milvus, collection_name, data_type, dimension, size, ni):
+        """ insert data to collection before testing """
         total_time = 0.0
         rps = 0.0
         ni_time = 0.0