diff --git a/infra/k8s/argo/compile-and-train-template.yaml b/infra/k8s/argo/compile-and-train-template.yaml
index 07d4e10c7..89b4d7e61 100644
--- a/infra/k8s/argo/compile-and-train-template.yaml
+++ b/infra/k8s/argo/compile-and-train-template.yaml
@@ -39,6 +39,8 @@ spec:
       # ── Compile ──
       - name: commit-sha
         value: HEAD
+      - name: git-branch
+        value: main
       - name: cuda-compute-cap
         value: "90"
       # ── Training ──