diff --git a/dockers/tpu-tests/Dockerfile b/dockers/tpu-tests/Dockerfile
index 530cc9ec6f..e23db55bb2 100644
--- a/dockers/tpu-tests/Dockerfile
+++ b/dockers/tpu-tests/Dockerfile
@@ -19,16 +19,16 @@ FROM pytorchlightning/pytorch_lightning:base-xla-py${PYTHON_VERSION}-torch${PYTO
 
 LABEL maintainer="Lightning-AI <https://github.com/Lightning-AI>"
 
-COPY ./ ./pytorch-lightning/
+COPY ./ ./lightning/
 
 # Pull the legacy checkpoints
-RUN cd pytorch-lightning && \
+RUN cd lightning && \
     bash .actions/pull_legacy_checkpoints.sh
 
 RUN \
     pip install -q fire && \
     # drop unnecessary packages
-    pip install -r pytorch-lightning/requirements/pytorch/devel.txt --no-cache-dir
+    pip install -r lightning/requirements/pytorch/devel.txt --no-cache-dir
 
 COPY ./dockers/tpu-tests/docker-entrypoint.sh /usr/local/bin/
 RUN chmod +x /usr/local/bin/docker-entrypoint.sh
diff --git a/dockers/tpu-tests/tpu_test_cases.jsonnet b/dockers/tpu-tests/tpu_test_cases.jsonnet
index 98056674cd..a73b25277f 100644
--- a/dockers/tpu-tests/tpu_test_cases.jsonnet
+++ b/dockers/tpu-tests/tpu_test_cases.jsonnet
@@ -22,7 +22,7 @@ local tputests = base.BaseTest {
     |||
       source ~/.bashrc
       conda activate lightning
-      mkdir -p /home/runner/work/pytorch-lightning && cd /home/runner/work/pytorch-lightning
+      mkdir -p /home/runner/work/lightning && cd /home/runner/work/lightning
       git clone https://github.com/Lightning-AI/lightning.git
       cd lightning
       echo $PWD