diff --git a/Dockerfile b/Dockerfile
index b79c223d..55967c8e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -119,8 +119,9 @@ RUN mkdir -p /aha/tmp/torch_install/
 # Save (and later restore) existing value for TMPDIR, if any
 ENV TMPTMPDIR=$TMPDIR
 ENV TMPDIR=/aha/tmp/torch_install/
-RUN source /aha/bin/activate && pip install --cache-dir=$TMPDIR --build=$TMPDIR torch==1.7.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
-RUN rm -rf $TMPDIR
+RUN source /aha/bin/activate && \
+  pip install --cache-dir=$TMPDIR --build=$TMPDIR torch==1.7.1+cpu -f https://download.pytorch.org/whl/torch_stable.html && \
+  rm -rf $TMPDIR
 ENV TMPDIR=$TMPTMPDIR
 
 WORKDIR /aha