diff --git a/test/images/node-perf/tf-wide-deep/Dockerfile b/test/images/node-perf/tf-wide-deep/Dockerfile
index e0e8bc617c1..a0d69002ef8 100644
--- a/test/images/node-perf/tf-wide-deep/Dockerfile
+++ b/test/images/node-perf/tf-wide-deep/Dockerfile
@@ -32,7 +32,9 @@ RUN wget https://github.com/tensorflow/models/archive/v1.9.0.tar.gz \
 && tar xzf v1.9.0.tar.gz \
 && rm -f v1.9.0.tar.gz
 
+RUN python /models-1.9.0/official/wide_deep/data_download.py
+
 WORKDIR $HOME/models-1.9.0/official/wide_deep
 ENV PYTHONPATH=$PYTHONPATH:$HOME/models-1.9.0
 
-ENTRYPOINT python ./data_download.py && python ./wide_deep.py
+ENTRYPOINT python ./wide_deep.py