diff --git a/.github/workflows/traindiffs_tests.yml b/.github/workflows/traindiffs_tests.yml
index a2fdcb453..382f0dfe1 100644
--- a/.github/workflows/traindiffs_tests.yml
+++ b/.github/workflows/traindiffs_tests.yml
@@ -3,7 +3,7 @@ name: Containerized Training Differences Tests Jax vs PyTorch
 on: 
   pull_request:
     branches:
-      - 'python311' 
+      - 'main' 
 
 jobs:
   build_and_push_docker_image: