diff --git a/helm_chart/HyperPodHelmChart/charts/training-operators/templates/Deployment/training-operator-kubeflow-Deployment.yaml b/helm_chart/HyperPodHelmChart/charts/training-operators/templates/Deployment/training-operator-kubeflow-Deployment.yaml
index 3ef20b51..07c2b24a 100644
--- a/helm_chart/HyperPodHelmChart/charts/training-operators/templates/Deployment/training-operator-kubeflow-Deployment.yaml
+++ b/helm_chart/HyperPodHelmChart/charts/training-operators/templates/Deployment/training-operator-kubeflow-Deployment.yaml
@@ -55,4 +55,9 @@ spec:
         securityContext:
           allowPrivilegeEscalation: false
       serviceAccountName: training-operator
-      terminationGracePeriodSeconds: 10
\ No newline at end of file
+      terminationGracePeriodSeconds: 10
+      tolerations:
+      - key: sagemaker.amazonaws.com/node-health-status
+        operator: "Equal"
+        value: "Unschedulable"
+        effect: "NoSchedule"
\ No newline at end of file