diff --git a/tests/benchmark/milvus_benchmark/client.py b/tests/benchmark/milvus_benchmark/client.py
index 46bd0fbe0d..1a64aa38d6 100644
--- a/tests/benchmark/milvus_benchmark/client.py
+++ b/tests/benchmark/milvus_benchmark/client.py
@@ -3,7 +3,7 @@ import pdb
 import random
 import logging
 import json
-import time, datetime
+import time
 import traceback
 from multiprocessing import Process
 from pymilvus import Milvus, DataType