diff --git a/requirements.txt b/requirements.txt
index 4c0159b..41c24fd 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,3 @@
 colorama==0.4.1
-pyyaml==5.1
+pyyaml==5.4
 scikit-learn==0.20.3
\ No newline at end of file
diff --git a/setup.py b/setup.py
index af8f0c5..1554d5a 100644
--- a/setup.py
+++ b/setup.py
@@ -16,7 +16,7 @@
     package_data={'deepswarm': ['../settings/default.yaml']},
     install_requires=[
         'colorama==0.4.1',
-        'pyyaml==5.1',
+        'pyyaml==5.4',
         'scikit-learn==0.20.3',
     ],
     classifiers=[