diff --git a/tensorflow/lite/micro/examples/hello_world/train.py b/tensorflow/lite/micro/examples/hello_world/train.py
index 59483acf04a..cdb9098b2a4 100644
--- a/tensorflow/lite/micro/examples/hello_world/train.py
+++ b/tensorflow/lite/micro/examples/hello_world/train.py
@@ -120,7 +120,7 @@ def train_model(epochs, x_values, y_values):
             verbose=2)
 
   if FLAGS.save_tf_model:
-    model.save(FLAGS.save_dir, save_format="tf")
+    model.save(os.path.join(FLAGS.save_dir, "model.keras"))
     logging.info("TF model saved to %s", FLAGS.save_dir)
 
   return model
@@ -138,4 +138,4 @@ def main(_):
 
 
 if __name__ == "__main__":
-  app.run(main)
\ No newline at end of file
+  app.run(main)