diff --git a/unity-command/entrypoint.sh b/unity-command/entrypoint.sh
index 02d65b9..a1e2033 100644
--- a/unity-command/entrypoint.sh
+++ b/unity-command/entrypoint.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-SERIAL=$1; EMAIL=$2; PASSWORD=$3; COMMAND=$4;  UNITY_BUILDER=$4
+SERIAL=$1; EMAIL=$2; PASSWORD=$3; COMMAND=$4;  UNITY_BUILDER=$5
 DEFAULT_ARGS="-quit -logFile -"
 
 rm -rf $HOME/.config/unity3d
diff --git a/unity/action.yaml b/unity/action.yaml
index bdba0d6..f5fc0bb 100644
--- a/unity/action.yaml
+++ b/unity/action.yaml
@@ -14,6 +14,9 @@ inputs:
   command:
     description: "Unity command to run."
     required: false
+  unityBuilder:
+    description: "Whether or not to use the UnityBuilder instead of a Unity command."
+    required: false
 runs:
   using: docker
   #This is a base "mock" image which is replaced by the correct image in the "unity-command" action at runtime.
@@ -23,3 +26,4 @@ runs:
   - ${{ inputs.email }}
   - ${{ inputs.password }}
   - ${{ inputs.command }}
+  - ${{ inputs.unityBuilder }}