diff --git a/.github/workflows/create-image.yaml b/.github/workflows/create-image.yaml
index 8334e14..6b7931c 100644
--- a/.github/workflows/create-image.yaml
+++ b/.github/workflows/create-image.yaml
@@ -36,16 +36,17 @@ jobs:
           password: ${{ secrets.DOCKERHUB_TOKEN }}
 
       - name: Get Unity Version
+        id: unity-version
         run: |
           UNITY_VERSION=$(grep ${{ matrix.platform }} versions.txt | cut -d' ' -f2)
-          echo "UNITY_VERSION=$UNITY_VERSION" >> $GITHUB_ENV
+          echo "UNITY_VERSION=$UNITY_VERSION" >> $GITHUB_OUTPUT
           echo "Unity version for ${{ matrix.platform }}: $UNITY_VERSION"
 
       - name: Docker Build
         run: .gitea/workflows/scripts/build-image.sh
         id: build-image
         env:
-          UNITY_VERSION: ${{ env.UNITY_VERSION }}
+          UNITY_VERSION: ${{ steps.unity-version.outputs.UNITY_VERSION }}
           UNITY_PLATFORM: ${{ matrix.platform }}
           IMAGE: "${{ vars.DOCKERHUB_USERNAME }}/unity-runner"
           GAMECI_VERSION: ${{ env.GAMECI_VERSION }}