diff --git a/tests/data/tensorflow_mnist/mnist_v2.py b/tests/data/tensorflow_mnist/mnist_v2.py
index bf1750b386..589eb0b80d 100644
--- a/tests/data/tensorflow_mnist/mnist_v2.py
+++ b/tests/data/tensorflow_mnist/mnist_v2.py
@@ -199,6 +199,7 @@ def main(args):
         if args.current_host == args.hosts[0]:
             ckpt_manager.save()
             net.save("/opt/ml/model/1.keras")
+        print("Saved model at /opt/ml/model/1.keras")
 
 
 if __name__ == "__main__":