diff --git a/plugins/vi-mode/vi-mode.plugin.zsh b/plugins/vi-mode/vi-mode.plugin.zsh
index 38e168ed..d975380d 100644
--- a/plugins/vi-mode/vi-mode.plugin.zsh
+++ b/plugins/vi-mode/vi-mode.plugin.zsh
@@ -34,12 +34,14 @@ function zle-keymap-select() {
   fi
 }; zle -N zle-keymap-select
 
+eval "override-vi-mode-$(declare -f zle-line-init)"
 zle-line-init () {
   if [ $TERM = "screen" ]; then
     echo -ne $VIM_INSERT_COLOR
   elif [ $TERM != "linux" ]; then
     echo -ne $VIM_INSERT_COLOR
   fi
+  override-vi-mode-zle-line-init
 }; zle -N zle-line-init
 
 autoload -U select-quoted