diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 4a0a7e23..d12425ec 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -33,6 +33,10 @@ jobs:
         chmod +x ./mount_rubikpi3.sh
         ./mount_rubikpi3.sh ${{ env.base_image }} ${{ env.script }} ${{ github.ref_name }}
 
+    - name: Compress built image
+      run: |
+        tar -I 'xz -T0' -cf photonvision_rubikpi3.tar.xz ${{ env.image_dir }} --checkpoint=10000 --checkpoint-action=echo='%T'
+
     - uses: actions/upload-artifact@v4.3.4
       with:
         name: photonvision_rubikpi3.tar.xz
diff --git a/mount_rubikpi3.sh b/mount_rubikpi3.sh
index 6f056263..5bf342a7 100644
--- a/mount_rubikpi3.sh
+++ b/mount_rubikpi3.sh
@@ -209,6 +209,4 @@ find photonvision_rubikpi3 -mindepth 1 -type d -empty -delete
 
 # Set output for later steps
 # Save the rootfs image path for later steps
-echo "rootfs_image=$ROOTFS_IMG" >> $GITHUB_ENV
-tar -I 'xz -T0' -cf photonvision_rubikpi3.tar.xz photonvision_rubikpi3 --checkpoint=10000 --checkpoint-action=echo='%T'
-
+echo "image_dir=$PWD/photonvision_rubikpi3" >> $GITHUB_ENV