diff --git a/pyproject.toml b/pyproject.toml
index 0429e33..58ef935 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -57,7 +57,7 @@ doc = [
     "sphinx-gallery",
 ]
 oryx = [
-    "oryx",
+    "oryx@git+https://github.com/jax-ml/oryx",
 ]
 
 [tool.pyink]