diff --git a/requirements.txt b/requirements.txt
index 5bf8678..56c835f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,7 +8,7 @@ opencv-python==4.1.0.25
 Pillow==8.2.0
 pybullet==2.6.5
 python-pcl
-PyYAML==5.1.2
+PyYAML==5.4
 scikit-learn==0.20.4
 scipy==1.2.2
 six==1.11.0
diff --git a/setup.py b/setup.py
index d518093..91d5914 100644
--- a/setup.py
+++ b/setup.py
@@ -35,7 +35,7 @@
         'opencv-python==4.1.0.25',
         'Pillow==8.2.0',
         'python-pcl',
-        'PyYAML==5.1.2',
+        'PyYAML==5.4',
         'scikit-learn==0.20.4',
         'scipy==1.2.2',
         'six==1.11.0',