diff --git a/unity-project/action.yaml b/unity-project/action.yaml
index 8d6c5f8..b0ea9ec 100644
--- a/unity-project/action.yaml
+++ b/unity-project/action.yaml
@@ -27,6 +27,10 @@ inputs:
   cacheVolume:
     description: "Name of the volume to cache the Library folder to."
     required: false
+  removeContainer:
+    description: "Remove the mock container after building."
+    required: false
+    default: "false"
 runs:
   using: "composite"
   steps:
@@ -55,6 +59,7 @@ runs:
       email: ${{ inputs.email }}
       password: ${{ inputs.password }}
       command: -projectPath ${{ inputs.projectPath }} -buildTarget ${{ steps.getTarget.outputs.buildTarget }} ${{ inputs.command }}
+      removeContainer: ${{ inputs.removeContainer }}
   - name: "Cache the Library folder."
     if: ${{ inputs.cacheVolume }}
     uses: act/common/docker/docker-cp@master