diff --git a/plugins/kuberay/workspace/train.yaml b/plugins/kuberay/workspace/train.yaml
index 793e2b3866dd38fcb1023abf8d22fb7f146d0597..a60cd9c3c7f4939fd64e4f2f2750f0cd1abb6abe 100644
--- a/plugins/kuberay/workspace/train.yaml
+++ b/plugins/kuberay/workspace/train.yaml
@@ -104,9 +104,7 @@
     - name: Execute training script
       shell: |
         kubectl exec -n {{ namespace }} {{ head_pod_name }} -- \
-          python -u /tmp/{{ script_with_vars_name }}
-      args:
-        executable: /bin/bash
+          bash -c "timeout 600s python -u /tmp/{{ script_with_vars_name }}"
       register: training_output
       changed_when: false