diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml
index f2905ca..e03d37f 100644
--- a/.github/workflows/pr-gpu.yaml
+++ b/.github/workflows/pr-gpu.yaml
@@ -21,9 +21,9 @@ jobs:
       fail-fast: false
       matrix:
         include:
-        - name: "python3.9-pytorch2.3.1-gpus1"
+        - name: "python3.11-pytorch2.3.1-gpus1"
           gpu_num: 1
-          python_version: 3.9
+          python_version: 3.11
           container: mosaicml/pytorch:2.3.1_cu121-python3.11-ubuntu20.04
         - name: "python3.11-pytorch2.3.1-gpus2"
           gpu_num: 2