diff --git a/pytorch.dockerfile b/pytorch.dockerfile
index c552d2f..97764c0 100644
--- a/pytorch.dockerfile
+++ b/pytorch.dockerfile
@@ -49,7 +49,7 @@ PYTORCH_BUILD_VERSION=$(echo "${PYTORCH_BRANCH}" | tr -d "v") python${PYTHON_VER
 # https://github.com/AngryLoki/pytorch/commit/38d6f115c65b67946de6765165ece0288c8416d4
 
 WORKDIR /home/rocm/pytorch/dist
-RUN cp $(ls -l *.whl | awk '{ print $9 }' | tail -n1 | tr -d '\n') /home/rocm/pytorch-rocm.whl
+RUN cp $(ls -l *.whl | awk '{ print $9 }' | tail -n1 | tr -d '\n') /home/rocm/torch.whl
 
 WORKDIR /home/rocm
 CMD [ "bash", "-l" ]