diff --git a/examples/TensorFlow/Keras/MNIST/keras_to_MDF.py b/examples/TensorFlow/Keras/MNIST/keras_to_MDF.py
index 244afaf8..14923af6 100644
--- a/examples/TensorFlow/Keras/MNIST/keras_to_MDF.py
+++ b/examples/TensorFlow/Keras/MNIST/keras_to_MDF.py
@@ -7,7 +7,7 @@
 from modelspec.utils import _val_info
 from modeci_mdf.execution_engine import EvaluableGraph
 
-exec(open("./keras_model.py").read())
+model = exec(open("./keras_model.py").read())
 
 # load the keras model
 # model = tf.keras.models.load_model("kr_N_model.keras")
diff --git a/examples/TensorFlow/Keras/MNIST/model_plot.png b/examples/TensorFlow/Keras/MNIST/model_plot.png
index 8e52a38f..2cdf7647 100644
Binary files a/examples/TensorFlow/Keras/MNIST/model_plot.png and b/examples/TensorFlow/Keras/MNIST/model_plot.png differ