1 files changed, 1 insertions, 1 deletions
diff --git a/vi-mode.sh b/vi-mode.sh
index d7f4e4f..7e94689 100644
--- a/vi-mode.sh
+++ b/vi-mode.sh
@@ -54,7 +54,7 @@ if [[ -n "$ZSH_VERSION" ]]; then
     bindkey -M menuselect 'l' vi-forward-char
 
     # Map 'jk' to Escape key in INSERT mode
-    bindkey -M insert 'jk' vi-cmd-mode
+    bindkey -M viins 'jk' vi-cmd-mode
 
 # Set vi-mode and key bindings for bash
 elif [[ -n "$BASH_VERSION" ]]; then