diff --git a/noxfile.py b/noxfile.py
index 3f3c630c18edebc50f50c612dcc0c42f092aed5c..33b118625b95fc344b1ccb9f61d1ab1c096e00e4 100644
--- a/noxfile.py
+++ b/noxfile.py
@@ -146,7 +146,7 @@ def testsuite_pystencils2(session: nox.Session, cupy_version: str | None):
         install_cupy(session, cupy_version, skip_if_no_cuda=True)
 
     session.install(
-        "git+https://i10git.cs.fau.de/pycodegen/pystencils.git@fhennig/patches-for-lbmpy"
+        "git+https://i10git.cs.fau.de/pycodegen/pystencils.git@v2.0-dev"
     )
     editable_install(session, ["alltrafos", "use_cython", "interactive", "tests"])