diff --git a/pyproject.toml b/pyproject.toml
index 3606616..7f75827 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -30,7 +30,7 @@ dependencies = [
   "jax>=0.4.35",
   "jaxlib>=0.4.35",
   "jaxtyping",
-  "klujax>=0.3.0",
+  "klujax>=0.3.1",
   "matplotlib",
   "natsort",
   "networkx",