diff --git a/src/widgets.zsh b/src/widgets.zsh
index 43cd974..56b0cca 100644
--- a/src/widgets.zsh
+++ b/src/widgets.zsh
@@ -168,7 +168,7 @@ _zsh_autosuggest_partial_accept() {
 	# instead of after it. We increment it by 1 to keep the following logic
 	# working even in this case.
 	CURSOR_POS=$CURSOR
-	if [ $KEYMAP = 'vicmd' ]; then
+	if [[ "$KEYMAP" = "vicmd" ]]; then
 		(( CURSOR_POS++ ))
 	fi
 
diff --git a/zsh-autosuggestions.zsh b/zsh-autosuggestions.zsh
index 96996dc..4aa85df 100644
--- a/zsh-autosuggestions.zsh
+++ b/zsh-autosuggestions.zsh
@@ -446,7 +446,7 @@ _zsh_autosuggest_partial_accept() {
 	# instead of after it. We increment it by 1 to keep the following logic
 	# working even in this case.
 	CURSOR_POS=$CURSOR
-	if [ $KEYMAP = 'vicmd' ]; then
+	if [[ "$KEYMAP" = "vicmd" ]]; then
 		(( CURSOR_POS++ ))
 	fi