diff --git a/deeprole/code/nn_train/requirements.txt b/deeprole/code/nn_train/requirements.txt
index 65f9146..ba65e8c 100644
--- a/deeprole/code/nn_train/requirements.txt
+++ b/deeprole/code/nn_train/requirements.txt
@@ -29,7 +29,7 @@ ptyprocess==0.6.0
 Pygments==2.3.1
 python-dateutil==2.8.0
 pytz==2019.1
-PyYAML==5.1
+PyYAML==5.4
 scandir==1.10.0
 scipy==1.2.1
 simplegeneric==0.8.1