diff --git a/entrypoint.sh b/entrypoint.sh
index ce08a3f..510bcaa 100755
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -46,7 +46,7 @@ fi
 OUTPUT=""
 if [[ -n "${PLUGIN_NAME:-}" ]]; then
 	OUTPUT="--output type=image"
-	if [[ "$1" =~ .*,.* ]]; then
+	if [[ "$PLUGIN_NAME" =~ .*,.* ]]; then
 		# https://github.com/moby/buildkit/issues/797#issuecomment-1561601104
 		OUTPUT=",\\\"name=$PLUGIN_NAME\\\""
 	else