diff --git a/.github/container/Dockerfile.maxtext b/.github/container/Dockerfile.maxtext
index 766bfd080..f7c94a1c5 100644
--- a/.github/container/Dockerfile.maxtext
+++ b/.github/container/Dockerfile.maxtext
@@ -18,13 +18,13 @@ EOF
 
 RUN <<"EOF" bash -ex -o pipefail
 echo "-e file://${SRC_PATH_MAXTEXT}" >> /opt/pip-tools.d/requirements-maxtext.in
-echo "-r ${SRC_PATH_MAXTEXT}/dependencies/requirements/base_requirements/requirements.txt" >> /opt/pip-tools.d/requirements-maxtext.in
+echo "-r ${SRC_PATH_MAXTEXT}/src/dependencies/requirements/base_requirements/requirements.txt" >> /opt/pip-tools.d/requirements-maxtext.in
 EOF
 
 RUN <<"EOF" bash -exuo pipefail
 # Avoid old tensorboard constraint from google-cloud-aiplatform[tensorboard].
 sed -i "/^cloud-accelerator-diagnostics/d" \
-  ${SRC_PATH_MAXTEXT}/dependencies/requirements/base_requirements/requirements.txt
+  ${SRC_PATH_MAXTEXT}/src/dependencies/requirements/base_requirements/requirements.txt
 
 # remove custom build hook that conflicts with our dependencies introduced in
 # https://github.com/AI-Hypercomputer/maxtext/commit/de886f9a918d17cfb025a353e7da2bc4d64c26b5