diff --git a/mnist/web-ui/Dockerfile b/mnist/web-ui/Dockerfile
index 36b55255..94fa4eeb 100644
--- a/mnist/web-ui/Dockerfile
+++ b/mnist/web-ui/Dockerfile
@@ -28,7 +28,7 @@ RUN pip --no-cache-dir install \
         h5py \
         ipykernel \
         numpy \
-        tensorflow \
+        tensorflow==1.7.0 \
         tensorflow-serving-api \
         flask \
         && \