diff --git a/.github/workflows/cloud-tpu-ci-nightly.yml b/.github/workflows/cloud-tpu-ci-nightly.yml
index d178bd0f4..b5b56c506 100644
--- a/.github/workflows/cloud-tpu-ci-nightly.yml
+++ b/.github/workflows/cloud-tpu-ci-nightly.yml
@@ -25,8 +25,8 @@ jobs:
       - uses: actions/checkout@93ea575cb5d8a053eaa0ac8fa3b40d7e05a33cc8 # ratchet:actions/checkout@v3
       - name: Install JAX test requirements
         run: |
-          pip install -r build/test-requirements.txt
-          pip install -r build/collect-profile-requirements.txt
+          pip install -U -r build/test-requirements.txt
+          pip install -U -r build/collect-profile-requirements.txt
       - name: Install JAX
         run: |
           pip uninstall -y jax jaxlib libtpu-nightly
diff --git a/build/collect-profile-requirements.txt b/build/collect-profile-requirements.txt
index 0f1ba7c8f..da25d4b6f 100644
--- a/build/collect-profile-requirements.txt
+++ b/build/collect-profile-requirements.txt
@@ -1,4 +1,4 @@
 tensorflow
 tensorboard-plugin-profile
 # Needed for the profile plugin to work without error
-protobuf<=3.20
+protobuf