diff --git a/.github/workflows/pytest-cpu.yaml b/.github/workflows/pytest-cpu.yaml
index 276d0b8..f36a156 100644
--- a/.github/workflows/pytest-cpu.yaml
+++ b/.github/workflows/pytest-cpu.yaml
@@ -55,7 +55,7 @@ on:
 jobs:
   pytest-cpu:
     timeout-minutes: 30
-    runs-on: ubuntu-latest
+    runs-on: linux-ubuntu-latest
     container: ${{ inputs.container }}
     steps:
     - name: Checkout Repo
diff --git a/.github/workflows/pytest-gpu.yaml b/.github/workflows/pytest-gpu.yaml
index 1a6902d..a7808d5 100644
--- a/.github/workflows/pytest-gpu.yaml
+++ b/.github/workflows/pytest-gpu.yaml
@@ -56,7 +56,7 @@ on:
 jobs:
   pytest-gpu:
     timeout-minutes: ${{ inputs.gha-timeout }}
-    runs-on: ubuntu-latest
+    runs-on: linux-ubuntu-latest
     env:
       MOSAICML_API_KEY: ${{ secrets.mcloud-api-key }}
     steps: