diff --git a/tests/benchmark/milvus_benchmark/main.py b/tests/benchmark/milvus_benchmark/main.py
index 824d7da626..d403f38219 100644
--- a/tests/benchmark/milvus_benchmark/main.py
+++ b/tests/benchmark/milvus_benchmark/main.py
@@ -16,6 +16,7 @@ from milvus_benchmark import config, utils
 from milvus_benchmark import parser
 # from scheduler import back_scheduler
 from logs import log
+from logs.log import global_params
 
 log.setup_logging()
 logger = logging.getLogger("milvus_benchmark.main")