diff --git a/tests/benchmark/milvus_benchmark/client.py b/tests/benchmark/milvus_benchmark/client.py
index 187b60089b..d7a8f273a2 100644
--- a/tests/benchmark/milvus_benchmark/client.py
+++ b/tests/benchmark/milvus_benchmark/client.py
@@ -4,7 +4,6 @@ import time
 import traceback
 from multiprocessing import Process
 from pymilvus import Milvus, DataType
-import numpy as np
 import utils as util
 import config
 from milvus_benchmark.runners import utils