diff --git a/examples/mnist/Dockerfile b/examples/mnist/Dockerfile
index 8b593553..2b4af62e 100644
--- a/examples/mnist/Dockerfile
+++ b/examples/mnist/Dockerfile
@@ -1,6 +1,6 @@
 #This container contains your model and any helper scripts specific to your model.
 # When building the image inside mnist.ipynb the base docker image will be overwritten
-FROM tensorflow/tensorflow:1.15.2-py3
+FROM tensorflow/tensorflow:2.2.2-py3
 
 ADD mnist.py /opt/mnist.py
 RUN chmod +x /opt/mnist.py