diff --git a/test/correctness/setup.py b/test/correctness/setup.py
index 0ec796c..2a673ad 100644
--- a/test/correctness/setup.py
+++ b/test/correctness/setup.py
@@ -35,7 +35,7 @@
         'pytest',
         'pymongo==3.6.1',
         'python-dateutil',
-        'PyYAML==4.2b4',
+        'PyYAML==5.4',
         'psutil',
         'coloredlogs==4.0.0'
     ],