diff --git a/tests/milvus_benchmark/requirements.txt b/tests/milvus_benchmark/requirements.txt
index 1285b4d2ba..328cff1eb4 100644
--- a/tests/milvus_benchmark/requirements.txt
+++ b/tests/milvus_benchmark/requirements.txt
@@ -1,6 +1,6 @@
 numpy==1.16.3
 pymilvus>=0.1.18
-pyyaml==3.12
+pyyaml==5.1
 docker==4.0.2
 tableprint==0.8.0
 ansicolors==1.1.8
\ No newline at end of file