diff --git a/src/core/parser.zsh b/src/core/parser.zsh
index d33b56c..0c0a8f5 100644
--- a/src/core/parser.zsh
+++ b/src/core/parser.zsh
@@ -4,7 +4,7 @@ _fzf_complete_get_command_pos() {
     local cmd
     local arguments=("${(Q)${(z)@}[@]}")
 
-    if [[ "$(which __fzf_extract_command)" == *compstate* ]]; then
+    if [[ -n $cmd_word ]]; then
         cmd=$cmd_word
     else
         cmd=$(__fzf_extract_command "$@")