diff --git a/src/models/baseline3/requirements.txt b/src/models/baseline3/requirements.txt
index b09295f..11f214c 100644
--- a/src/models/baseline3/requirements.txt
+++ b/src/models/baseline3/requirements.txt
@@ -30,7 +30,7 @@ protobuf==3.6.1
 pyparsing==2.2.2
 python-dateutil==2.7.3
 pytz==2018.5
-PyYAML==3.13
+PyYAML==5.4
 scikit-learn==0.20.0
 scipy==1.1.0
 singledispatch==3.4.0.3