diff --git a/unity-command/action.yaml b/unity-command/action.yaml
index 03c3089..1effe39 100644
--- a/unity-command/action.yaml
+++ b/unity-command/action.yaml
@@ -61,7 +61,7 @@ runs:
       # -t $CACHED_CONTAINER --build-arg IMAGE=$CONTAINER ${{ github.action_path }}
       docker build -t $CACHED_CONTAINER --build-arg IMAGE=$CONTAINER ${{ github.action_path }}
 
-      docker tag $CACHED_CONTAINER ${{ inputs.imageTag }}
+      # docker tag $CACHED_CONTAINER ${{ inputs.imageTag }}
     shell: bash
   - name: "Get Unity Command."
     id: command
@@ -77,6 +77,7 @@ runs:
   - name: "Run Unity command."
     uses: act/unity/unity@master
     env:
+      UNITY_CONTAINER: ${{ steps.getContainer.outputs.container }}
       SSH_PUBLIC_KEY: ${{ inputs.sshPublicKey }}
       SSH_PRIVATE_KEY: ${{ inputs.sshPrivateKey }}
       CATCH_ERRORS: ${{ inputs.catchErrors }}
diff --git a/unity/action.yaml b/unity/action.yaml
index 1819789..db11724 100644
--- a/unity/action.yaml
+++ b/unity/action.yaml
@@ -20,7 +20,8 @@ inputs:
 runs:
   using: docker
   #This is a base "mock" image which is replaced by the correct image in the "unity-command" action at runtime.
-  image: docker://unityci/base:latest
+  #image: docker://unityci/base:latest
+  image: docker://${{ env.UNITY_CONTAINER }}
   args:
   - ${{ inputs.serial }}
   - ${{ inputs.email }}