diff --git a/pyproject.toml b/pyproject.toml
index a204633..7eee392 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,7 +11,7 @@ readme = "README.md"
 requires-python = ">=3.9"
 license = { file = "LICENSE" }
 urls = { "Homepage" = "https://github.com/DifferentiableUniverseInitiative/JaxPM" }
-dependencies = ["jax_cosmo", "jax>=0.4.35", "jaxdecomp>=0.2.3"]
+dependencies = ["jax_cosmo", "jax>=0.4.35", "jaxdecomp>=0.2.4"]
 
 [tool.setuptools]
 packages = ["jaxpm"]