diff --git a/pyproject.toml b/pyproject.toml
index 69c0833..9fb84d2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,14 +11,14 @@ readme = "README.md"
 requires-python = ">=3.9"
 license = { file = "LICENSE" }
 urls = { "Homepage" = "https://github.com/DifferentiableUniverseInitiative/JaxPM" }
-dependencies = ["jax_cosmo", "jax>=0.4.30", "jaxdecomp>=0.2.1"]
+dependencies = ["jax_cosmo", "jax>=0.4.30", "jaxdecomp>=0.2.2"]
 
 [project.optional-dependencies]
 test = [
     "jax>=0.4.30",
     "numpy",
     "jax_cosmo",
-    "jaxdecomp>=0.2.1",
+    "jaxdecomp>=0.2.2",
     "pytest>=8.0.0",
     "pfft-python @ git+https://github.com/MP-Gadget/pfft-python",
     "pmesh @ git+https://github.com/MP-Gadget/pmesh",