diff --git a/setup.sh b/setup.sh
index 3c3a0d14b..c1c96213b 100644
--- a/setup.sh
+++ b/setup.sh
@@ -153,7 +153,7 @@ if [[ "$MODE" == "nightly" ]]; then
     echo "-------------------------------------------------"
     
     python3 -m uv pip install --no-cache-dir -U -r requirements.txt.nightly-temp \
-                                                -r "${MAXTEXT_REPO_ROOT?}"'/extra_deps_from_github.txt'
+                                                -r "${MAXTEXT_REPO_ROOT?}"'/src/install_maxtext_extra_deps/extra_deps_from_github.txt'
     rm requirements.txt.nightly-temp
 else
     # stable or stable_stack mode: Install with pinned commits
@@ -172,7 +172,7 @@ else
       exit 2
     else
       python3 -m uv pip install --resolution=lowest -r "$tpu_requirements_txt" \
-                                                    -r "${MAXTEXT_REPO_ROOT?}"'/extra_deps_from_github.txt'
+                                                    -r "${MAXTEXT_REPO_ROOT?}"'/src/install_maxtext_extra_deps/extra_deps_from_github.txt'
     fi
 fi