1 files changed, 2 insertions, 1 deletions
diff --git a/vi-mode.sh b/vi-mode.sh
index 7776106..5c3678f 100644
--- a/vi-mode.sh
+++ b/vi-mode.sh
@@ -31,9 +31,10 @@ if [[ -n "$ZSH_VERSION" ]]; then
         PS1="%{$terminfo_down_sc$VI_MODE$terminfo[rc]%}%~ $ "
     }
 
-    function zle-line-init() zle-keymap-select() {
+    function zle-line-init () {
         set-prompt
         zle reset-prompt
+        zle-keymap-select
     }
     preexec () { print -rn -- $terminfo[el]; }