diff --git a/setup.cfg b/setup.cfg
index 7bf91ef..67c5e33 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -47,7 +47,7 @@ pyspod.utils.plotting_support =
 
 [options.extras_require]
 ai = tensorflow
-mpi = mpi4py
+mpi = mpi4py >= 3.1
 test =
     pytest
     pytest-cov