diff --git a/action.yml b/action.yml
index 7c857d6..8a3d789 100644
--- a/action.yml
+++ b/action.yml
@@ -10,3 +10,5 @@ runs:
   using: 'docker'
   image: 'Dockerfile'
   entrypoint: './entrypoint.sh'
+  env:
+    OUTPUT_FOLDER: ${{ inputs.output-folder }}
diff --git a/entrypoint.sh b/entrypoint.sh
index f7bb393..43b6457 100755
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -6,7 +6,7 @@ set -e
 just build
 
 # Use the correct environment variable for the input
-output_folder="${INPUT_OUTPUT-FOLDER}"
+output_folder="${OUTPUT_FOLDER}"
 
 # Ensure the output folder exists
 mkdir -p "$output_folder"