diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 1e570fce0..9e95485f7 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -104,5 +104,5 @@ jobs:
           push: ${{ github.event_name != 'pull_request' }}
           tags: |
              ${{ steps.metadata.outputs.tags }}-py${{ matrix.python_version }}-cu${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
-             ${{ (matrix.is_latest) && format('{0}:latest', steps.metadata.outputs.image) || '' }}
+             ${{ (matrix.is_latest) && format('{0}:latest', steps.metadata.inputs.images[0]) || '' }}
           labels: ${{ steps.metadata.outputs.labels }}