diff --git a/tests/benchmark/milvus_benchmark/runners/base.py b/tests/benchmark/milvus_benchmark/runners/base.py
index 589ad4d10c..cd94a44990 100644
--- a/tests/benchmark/milvus_benchmark/runners/base.py
+++ b/tests/benchmark/milvus_benchmark/runners/base.py
@@ -1,7 +1,6 @@
 import time
 import pdb
 import logging
-import threading
 import traceback
 import grpc
 import numpy as np