diff --git a/example/pytorch/torchelastic/torchelastic.yaml b/example/pytorch/torchelastic/torchelastic.yaml
index 938be11e..9a3c5b18 100644
--- a/example/pytorch/torchelastic/torchelastic.yaml
+++ b/example/pytorch/torchelastic/torchelastic.yaml
@@ -23,7 +23,7 @@ spec:
               imagePullPolicy: Always
     Worker:
       replicas: 1
-      restartPolicy: ExitCode
+      restartPolicy: OnFailure
       template:
         spec:
           volumes: