diff --git a/.github/workflows/ci-nightly-build-test.yaml b/.github/workflows/ci-nightly-build-test.yaml
index 845c20afe..319dc46dc 100644
--- a/.github/workflows/ci-nightly-build-test.yaml
+++ b/.github/workflows/ci-nightly-build-test.yaml
@@ -100,7 +100,7 @@ jobs:
     needs: Decision
     container:
       image: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:infrastructure-public-image-82c7ac59ba4b
-    runs-on: linux-x86-n2-32
+    runs-on: linux-x86-n4-32
     steps:
       - name: Check out a copy of the TFQ git repository
         uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0 # v7.0.0
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index f83bc8c69..345ceb397 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -92,7 +92,7 @@ jobs:
 
   bazel-tests:
     name: Library tests
-    runs-on: linux-x86-n2-32
+    runs-on: linux-x86-n4-32
     container:
       image: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:infrastructure-public-image-82c7ac59ba4b
     timeout-minutes: 60
@@ -145,7 +145,7 @@ jobs:
 
   tutorials-test:
     name: Tutorial tests
-    runs-on: linux-x86-n2-32
+    runs-on: linux-x86-n4-32
     container:
       image: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:infrastructure-public-image-82c7ac59ba4b
     timeout-minutes: 60