diff --git a/AttentiveRanker/src/requirements.txt b/AttentiveRanker/src/requirements.txt
index 81d3454..457629d 100644
--- a/AttentiveRanker/src/requirements.txt
+++ b/AttentiveRanker/src/requirements.txt
@@ -11,7 +11,7 @@ mock==3.0.5
 numpy==1.15.4
 pkg-resources==0.0.0
 protobuf==3.9.1
-PyYAML==5.1.2
+PyYAML==5.4
 scipy==1.3.1
 six==1.12.0
 tensorboard==1.13.1