diff --git a/plugin.sh b/plugin.sh
index 03be9e6..0a479c2 100755
--- a/plugin.sh
+++ b/plugin.sh
@@ -60,14 +60,14 @@ if [ -n "${PLUGIN_BUILD_ARGS_FROM_ENV:-}" ]; then
     BUILD_ARGS_FROM_ENV=$(echo "${PLUGIN_BUILD_ARGS_FROM_ENV}" | tr ',' '\n' | while read build_arg; do echo "--build-arg ${build_arg}=$(eval "echo \$$build_arg")"; done)
 fi
 
-# if [[ "${PLUGIN_SKIP_UNUSED_STAGES:-}" == "true" ]]; then
-#     SKIP_UNUSED_STAGES="--skip-unused-stages=true"
-# fi
-
-if [[ -n "${PLUGIN_SNAPSHOT_MODE:-}" ]]; then
-    SNAPSHOT_MODE="--snapshot-mode=${PLUGIN_SNAPSHOT_MODE}"
+if [[ "${PLUGIN_SKIP_UNUSED_STAGES:-}" == "true" ]]; then
+    SKIP_UNUSED_STAGES="--skip-unused-stages=true"
 fi
 
+# if [[ -n "${PLUGIN_SNAPSHOT_MODE:-}" ]]; then
+#     SNAPSHOT_MODE="--snapshot-mode=${PLUGIN_SNAPSHOT_MODE}"
+# fi
+
 # auto_tag, if set auto_tag: true, auto generate .tags file
 # support format Major.Minor.Release or start with `v`
 # docker tags: Major, Major.Minor, Major.Minor.Release and latest
@@ -116,7 +116,8 @@ fi
     ${CACHE_TTL:-} \
     ${CACHE_REPO:-} \
     ${TARGET:-} \
-    ${SNAPSHOT_MODE:-} \
+    ${SKIP_UNUSED_STAGES:-} \
+
     ${BUILD_ARGS:-} \
     ${BUILD_ARGS_FROM_ENV:-}
     
\ No newline at end of file