diff --git a/tensorflow-notebook/Dockerfile b/tensorflow-notebook/Dockerfile
index c14d8f40..fb735b90 100644
--- a/tensorflow-notebook/Dockerfile
+++ b/tensorflow-notebook/Dockerfile
@@ -9,16 +9,16 @@ LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"
 # Fix DL4006
 SHELL ["/bin/bash", "-o", "pipefail", "-c"]
 
-# Temporary fix for https://github.com/jupyter/docker-stacks/issues/1480
-USER root
-WORKDIR /opt/conda/lib/python3.9/site-packages/
-RUN wget -qO- https://patch-diff.githubusercontent.com/raw/tensorflow/tensorflow/pull/51450.diff | git apply
-USER ${NB_UID}
-WORKDIR "${HOME}"
-
 # Install Tensorflow
 RUN mamba install --quiet --yes \
     'tensorflow' && \
     mamba clean --all -f -y && \
     fix-permissions "${CONDA_DIR}" && \
     fix-permissions "/home/${NB_USER}"
+
+# Temporary fix for https://github.com/jupyter/docker-stacks/issues/1480
+USER root
+WORKDIR /opt/conda/lib/python3.9/site-packages/
+RUN wget -qO- https://patch-diff.githubusercontent.com/raw/tensorflow/tensorflow/pull/51450.diff | git apply
+USER ${NB_UID}
+WORKDIR "${HOME}"