diff --git a/.github/workflows/cloud-tpu-presubmit.yml b/.github/workflows/cloud-tpu-presubmit.yml
index 406de87aee06..ac595afdabbb 100644
--- a/.github/workflows/cloud-tpu-presubmit.yml
+++ b/.github/workflows/cloud-tpu-presubmit.yml
@@ -35,6 +35,8 @@ jobs:
         run: |
           pip install -U -r build/test-requirements.txt
       # TODO: build jax should be done on a step prior or we should just bazel test
+      - name: Wait For Connection
+        uses: google-ml-infra/jax-fork/actions/ci_connection@28deea80eeffa1e98ba84ebb29200b7b00c1b61f
       - name: Build JAX
         run: |
           pip uninstall -y jaxlib