diff --git a/video_prediction_tools/env_setup/requirements.txt b/video_prediction_tools/env_setup/requirements.txt
index a831d7d8daf0159ad3e1c5709db9bb9a71cf5779..9f433734a966541c6c6a20a6387a499716b2d80a 100755
--- a/video_prediction_tools/env_setup/requirements.txt
+++ b/video_prediction_tools/env_setup/requirements.txt
@@ -1,5 +1,5 @@
 matplotlib==3.3.0
-#mpi4py==3.0.3
+mpi4py==3.0.1
 pandas==0.25.3
 xarray==0.16.0
 basemap==1.3.0