diff --git a/pyproject.toml b/pyproject.toml
index f157307..5715787 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,7 +25,7 @@ dependencies = [
     "dm-haiku>=0.0.9",
     "flax>=0.6.3",
     "optax>=0.1.3",
-    "surjectors@git+https://git@github.com/dirmeier/surjectors@v0.2.2",
+    "surjectors>=0.2.2",
 ]
 dynamic = ["version"]