diff --git a/infra/k8s/argo/gpu-test-pipeline-template.yaml b/infra/k8s/argo/gpu-test-pipeline-template.yaml
index 381e485b3..5d5afc781 100644
--- a/infra/k8s/argo/gpu-test-pipeline-template.yaml
+++ b/infra/k8s/argo/gpu-test-pipeline-template.yaml
@@ -131,6 +131,8 @@ spec:
             value: /data/test-data
           - name: RUST_LOG
             value: info
+          - name: LD_LIBRARY_PATH
+            value: /usr/local/nvidia/lib64:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64
         volumeMounts:
           - name: git-ssh-key
             mountPath: /etc/git-ssh
@@ -179,6 +181,8 @@ spec:
                 echo "=== Updating checkout ==="
                 git fetch origin
                 git checkout --force "$REF"
+                # If REF is a branch, fast-forward to origin
+                git reset --hard "origin/$REF" 2>/dev/null || true
                 git clean -fd
               fi
             else