diff --git a/tests/benchmark/milvus_benchmark/main.py b/tests/benchmark/milvus_benchmark/main.py
index 829cc4b851..d93fa389ec 100644
--- a/tests/benchmark/milvus_benchmark/main.py
+++ b/tests/benchmark/milvus_benchmark/main.py
@@ -32,6 +32,7 @@ def positive_int(s):
 
 
 def get_image_tag(image_version):
+    """ Set the image version to the latest version """
     return "%s-latest" % (image_version)