From c69d21016caa712481f3738a62ba2722d5583047 Mon Sep 17 00:00:00 2001
From: Matthew Hoener <matthewh@ca.ibm.com>
Date: Wed, 7 Feb 2018 14:56:48 -0800
Subject: [PATCH] Fix vi mode plugin

---
 powerlevel9k.zsh-theme | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/powerlevel9k.zsh-theme b/powerlevel9k.zsh-theme
index 14d7bd81..15a0b68a 100755
--- a/powerlevel9k.zsh-theme
+++ b/powerlevel9k.zsh-theme
@@ -1522,6 +1522,11 @@ NEWLINE='
   [[ $POWERLEVEL9K_PROMPT_ADD_NEWLINE == true ]] && PROMPT="$NEWLINE$PROMPT"
 }
 
+zle-keymap-select () {
+	zle reset-prompt
+	zle -R
+}
+
 prompt_powerlevel9k_setup() {
   # The value below was set to better support 32-bit CPUs.
   # It's the maximum _signed_ integer value on 32-bit CPUs.
@@ -1590,6 +1595,8 @@ prompt_powerlevel9k_setup() {
   # prepare prompts
   add-zsh-hook precmd powerlevel9k_prepare_prompts
   add-zsh-hook preexec powerlevel9k_preexec
+
+  zle -N zle-keymap-select
 }
 
 prompt_powerlevel9k_teardown() {