diff --git a/tests/benchmark/milvus_benchmark/utils.py b/tests/benchmark/milvus_benchmark/utils.py
index 3aab66cf75..b9447526ac 100644
--- a/tests/benchmark/milvus_benchmark/utils.py
+++ b/tests/benchmark/milvus_benchmark/utils.py
@@ -4,10 +4,10 @@ import logging
 import string
 import random
 from yaml.representer import SafeRepresenter
-from yaml import full_load, dump
+# from yaml import full_load, dump
 import yaml
 import tableprint as tp
-from pprint import pprint
+# from pprint import pprint
 import config
 
 logger = logging.getLogger("milvus_benchmark.utils")