diff --git a/smdebug/tensorflow/keras.py b/smdebug/tensorflow/keras.py
index 37d6e61db..53ab947b3 100644
--- a/smdebug/tensorflow/keras.py
+++ b/smdebug/tensorflow/keras.py
@@ -390,7 +390,7 @@ def _get_exec_function(self, mode):
     def _validate_exec_function(self, fn):
         if fn is None:
             self.logger.info(
-                f"Could not save tensors for {self.mode} step {self.mode_steps[self.mode]} "
+                f"Could not save tensors for mode {self.mode.name} step {self.mode_steps[self.mode]} "
                 f"as execution function has not yet been built."
             )
             return False
diff --git a/tests/tensorflow/hooks/test_training_end.py b/tests/tensorflow/hooks/test_training_end.py
index 309ec42af..541b6be56 100644
--- a/tests/tensorflow/hooks/test_training_end.py
+++ b/tests/tensorflow/hooks/test_training_end.py
@@ -23,6 +23,8 @@ def test_training_job_has_ended(out_dir):
             "10",
             "--save_frequency",
             "5",
+            "--script-mode",
+            "y",
         ],
         env={"CUDA_VISIBLE_DEVICES": "-1", "SMDEBUG_LOG_LEVEL": "debug"},
     )