diff --git a/tests/benchmark/milvus_benchmark/update.py b/tests/benchmark/milvus_benchmark/update.py
index e986c10670..c060105c3f 100644
--- a/tests/benchmark/milvus_benchmark/update.py
+++ b/tests/benchmark/milvus_benchmark/update.py
@@ -6,6 +6,8 @@ import argparse
 from yaml import full_load, dump
 import config
 import utils
+import requests
+import json
 
 
 def parse_server_tag(server_tag):