diff --git a/tests/benchmark/milvus_benchmark/client.py b/tests/benchmark/milvus_benchmark/client.py
index fe46c331ef..2271508af8 100644
--- a/tests/benchmark/milvus_benchmark/client.py
+++ b/tests/benchmark/milvus_benchmark/client.py
@@ -490,6 +490,12 @@ class MilvusClient(object):
             collection_name = self._collection_name
         return self._milvus.release_partitions(collection_name, tag_names, timeout=timeout)
 
+    @time_wrapper
+    def get_query_segment_info(self, collection_name=None, timeout=300, **kwargs):
+        if collection_name is None:
+            collection_name = self._collection_name
+        return self._milvus.get_query_segment_info(collection_name, timeout=timeout, **kwargs)
+
     @time_wrapper
     def scene_test(self, collection_name=None, vectors=None, ids=None):
         """