diff --git a/main.go b/main.go
index d1f2a19..b5e4a7e 100644
--- a/main.go
+++ b/main.go
@@ -85,7 +85,7 @@ func main() {
 		case "fish":
 			fmt.Printf(`function _llm_shell_hint
     set -l LLM_SUGGESTION_FILE (mktemp)
-    llm-shell-hint %s query --output $LLM_SUGGESTION_FILE (commandline -b)
+    llm-shell-hint %s --output $LLM_SUGGESTION_FILE query (commandline -b)
     set -l LLM_SUGGESTION (cat $LLM_SUGGESTION_FILE)
     rm $LLM_SUGGESTION_FILE
 
@@ -102,7 +102,7 @@ bind ctrl-q _llm_shell_hint
 		case "bash":
 			fmt.Printf(`_llm_shell_hint() {
     local suggestion_file=$(mktemp)
-    llm-shell-hint %s query --output "$suggestion_file" -- "$READLINE_LINE"
+    llm-shell-hint %s --output "$suggestion_file" query -- "$READLINE_LINE"
     local suggestion=$(<"$suggestion_file")
     rm "$suggestion_file"