diff --git a/src/ml/requirements.txt b/src/ml/requirements.txt
index 7872545..c576639 100644
--- a/src/ml/requirements.txt
+++ b/src/ml/requirements.txt
@@ -17,7 +17,7 @@ protobuf==3.5.2.post1
 pyparsing==2.2.0
 python-dateutil==2.7.3
 pytz==2018.4
-PyYAML==3.12
+PyYAML==5.4
 scikit-learn==0.19.1
 scipy==1.1.0
 six==1.11.0