diff --git a/desktop/piecannon.in b/desktop/piecannon.in
index 192b125..68b1ec1 100644
--- a/desktop/piecannon.in
+++ b/desktop/piecannon.in
@@ -1,7 +1,7 @@
 #!/bin/sh
 JAR_PATH="@JAR_PATH@"
 
-if [ $1 == "--prompt" ]; then
+if [ "$1" = "--file-selector" ]; then
     RESULT=$(java -jar "$JAR_PATH" $(yad --file))
 else
     RESULT=$(java -jar "$JAR_PATH" $@)