diff --git a/infra/ansible/Dockerfile b/infra/ansible/Dockerfile
index 53ef91f9432..f2f327044ca 100644
--- a/infra/ansible/Dockerfile
+++ b/infra/ansible/Dockerfile
@@ -20,8 +20,7 @@ ARG ansible_vars
 RUN ansible-playbook -vvv playbook.yaml -e "stage=release" -e "${ansible_vars}" --tags "install_deps"
 
 WORKDIR /tmp/wheels
-COPY --from=build /src/pytorch/dist/*.whl ./
-COPY --from=build /src/pytorch/xla/dist/*.whl ./
+COPY --from=build /dist/*.whl ./
 
 RUN echo "Installing the following wheels" && ls *.whl
 RUN pip install *.whl