diff --git a/noxfile.py b/noxfile.py
index 7192977..a40999e 100644
--- a/noxfile.py
+++ b/noxfile.py
@@ -16,7 +16,7 @@ def lint(session: nox.Session) -> None:
 @nox.session
 @nox.parametrize("group", ["bionty-unit", "bionty-docs"])
 def build(session: nox.Session, group: str):
-    branch = "migrate" if IS_PR else "main"  # point back to "release"
+    branch = "main" if IS_PR else "main"  # point back to "release"
     install_lamindb(session, branch=branch)
     session.run(*"uv pip install --system -e .[dev]".split())