diff --git a/tensorflow/Dockerfile b/tensorflow/Dockerfile
index 1ac997501adf8e5406d0df2bf02bb6caf0539f69..96b4410e4bc09d374e2684f1ab3e3cb4bd02195c 100644
--- a/tensorflow/Dockerfile
+++ b/tensorflow/Dockerfile
@@ -41,8 +41,8 @@ RUN mkdir -p /opt/tensorflow/.venv && \
     python3.7 -m venv /opt/tensorflow/.venv/python3.7 && \
     . /opt/tensorflow/.venv/python3.7/bin/activate && \
     pip --no-cache-dir install --upgrade pip && \
+    pip --no-cache-dir install wheel && \
     pip --no-cache-dir install \
-        wheel \
         Pillow \
         h5py \
         keras_preprocessing \