diff --git a/onnxruntime/test/python/onnxruntime_test_python.py b/onnxruntime/test/python/onnxruntime_test_python.py
index 59dbfd791b..7a2c46c66a 100644
--- a/onnxruntime/test/python/onnxruntime_test_python.py
+++ b/onnxruntime/test/python/onnxruntime_test_python.py
@@ -907,10 +907,10 @@ class TestInferenceSession(unittest.TestCase):
             return
 
         # Exclude for training
-        training_enabled = false
+        training_enabled = False
         try:
             from onnxruntime.capi.ort_trainer import ORTTrainer
-            training_enabled = true
+            training_enabled = True
         except:
             pass