diff --git a/.github/workflows/coverage.yaml b/.github/workflows/coverage.yaml
index 5ca86861fb..75643e7a57 100644
--- a/.github/workflows/coverage.yaml
+++ b/.github/workflows/coverage.yaml
@@ -9,6 +9,7 @@ jobs:
   coverage:
     timeout-minutes: 5
     runs-on: ubuntu-latest
+    container: mosaicml/pytorch:2.4.0_cpu-python3.11-ubuntu20.04
     steps:
     - name: Checkout Repo
       uses: actions/checkout@v3