diff --git a/tests/benchmark/milvus_benchmark/runners/locust_task.py b/tests/benchmark/milvus_benchmark/runners/locust_task.py
index 4b79b24f0b..ba54242a57 100644
--- a/tests/benchmark/milvus_benchmark/runners/locust_task.py
+++ b/tests/benchmark/milvus_benchmark/runners/locust_task.py
@@ -1,6 +1,5 @@
 import time
 import pdb
-import random
 import logging
 from locust import User, events
 from milvus_benchmark.client import MilvusClient