diff --git a/test/images/node-perf/tf-wide-deep/Dockerfile b/test/images/node-perf/tf-wide-deep/Dockerfile
index 37eac1e6dd1..e0e8bc617c1 100644
--- a/test/images/node-perf/tf-wide-deep/Dockerfile
+++ b/test/images/node-perf/tf-wide-deep/Dockerfile
@@ -21,10 +21,10 @@ RUN apt-get update && apt-get install -y wget time
 
 RUN case $(uname -m) in \
     aarch64) \
-      pip install tensorflow-aarch64; \
+      pip install tensorflow-aarch64==1.2; \
       ;; \
     *) \
-      pip install tensorflow; \
+      pip install tensorflow==1.9.0; \
       ;; \
     esac
 
@@ -35,4 +35,4 @@ RUN wget https://github.com/tensorflow/models/archive/v1.9.0.tar.gz \
 WORKDIR $HOME/models-1.9.0/official/wide_deep
 ENV PYTHONPATH=$PYTHONPATH:$HOME/models-1.9.0
 
-ENTRYPOINT python ./wide_deep.py
+ENTRYPOINT python ./data_download.py && python ./wide_deep.py
diff --git a/test/images/node-perf/tf-wide-deep/README.md b/test/images/node-perf/tf-wide-deep/README.md
index d69ffac20d2..4df11c85209 100644
--- a/test/images/node-perf/tf-wide-deep/README.md
+++ b/test/images/node-perf/tf-wide-deep/README.md
@@ -2,7 +2,7 @@
 
 The container image described here predicts the income using the census income dataset in Tensorflow. For more
 information, see
-[https://github.com/tensorflow/models/tree/master/official/r1/wide_deep](https://github.com/tensorflow/models/tree/master/official/r1/wide_deep).
+[https://github.com/tensorflow/models/tree/v2.0/official/r1/wide_deep](https://github.com/tensorflow/models/tree/v2.0/official/r1/wide_deep).
 This image is used as a workload in node performance testing.
 
 ## How to release:
diff --git a/test/images/node-perf/tf-wide-deep/VERSION b/test/images/node-perf/tf-wide-deep/VERSION
index 9459d4ba2a0..5625e59da88 100644
--- a/test/images/node-perf/tf-wide-deep/VERSION
+++ b/test/images/node-perf/tf-wide-deep/VERSION
@@ -1 +1 @@
-1.1
+1.2