diff --git a/tests/milvus_benchmark/requirements.txt b/tests/milvus_benchmark/requirements.txt
index 1285b4d2bafe32adb372e6cad77852dc1b1f1948..328cff1eb42901b5212369f669bd99382db6c16c 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