diff --git a/.ci_support/environment-mpich.yml b/.ci_support/environment-mpich.yml
index e3b084a6..3f71d5c1 100644
--- a/.ci_support/environment-mpich.yml
+++ b/.ci_support/environment-mpich.yml
@@ -10,4 +10,4 @@ dependencies:
   - distributed
   - dask-jobqueue
   - mpi4py =3.1.4
-  - pympipool =0.5.2
\ No newline at end of file
+  - pympipool =0.5.4
\ No newline at end of file
diff --git a/.ci_support/environment-openmpi.yml b/.ci_support/environment-openmpi.yml
index 368a3d66..43468178 100644
--- a/.ci_support/environment-openmpi.yml
+++ b/.ci_support/environment-openmpi.yml
@@ -10,4 +10,4 @@ dependencies:
   - distributed 
   - dask-jobqueue
   - mpi4py =3.1.4
-  - pympipool =0.5.2
\ No newline at end of file
+  - pympipool =0.5.4
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 20ca6891..729ec19e 100644
--- a/setup.py
+++ b/setup.py
@@ -26,7 +26,7 @@
     keywords='lammps, mpi4py',
     packages=find_packages(exclude=["*tests*"]),
     install_requires=[
-        "mpi4py==3.1.4", "pympipool==0.5.2",
+        "mpi4py==3.1.4", "pympipool==0.5.4",
     ],
     cmdclass=versioneer.get_cmdclass(),
 )