diff --git a/infra/k8s/argo/train-dqn-template.yaml b/infra/k8s/argo/train-dqn-template.yaml
index cc7c4203c..34a2a39ce 100644
--- a/infra/k8s/argo/train-dqn-template.yaml
+++ b/infra/k8s/argo/train-dqn-template.yaml
@@ -32,7 +32,9 @@ spec:
     parameters:
       - name: model
         value: dqn
-      - name: commit-ref
+      - name: commit-sha
+        value: HEAD
+      - name: git-branch
         value: main
       - name: gpu-pool
         value: ci-training-h100