diff --git a/.config/zsh/theme b/.config/zsh/theme
index 9de1e7b..fbc2688 100644
--- a/.config/zsh/theme
+++ b/.config/zsh/theme
@@ -23,10 +23,19 @@ zstyle ':vcs_info:*' enable git
 	fi
 }
 
+function zle-line-init zle-keymap-select {
+	PS1="
+%B%F{blue}%~ \$vcs_info_msg_0_
+%F{green}${${KEYMAP/vicmd/N }/(main|viins)/}$>%f%b "
+	PS2=$PS1
+    # RPS1="${${KEYMAP/vicmd/-- NORMAL --}/(main|viins)/-- INSERT --}"
+    # RPS2=$RPS1
+	zle reset-prompt
+}
+zle -N zle-line-init
+zle -N zle-keymap-select
+
 precmd () { vcs_info }
 #RPROMPT=\$vcs_info_msg_0_
-PS1="
-%B%F{blue}%~ \$vcs_info_msg_0_
-%F{green}$>%f%b "
 ZSH_AUTOSUGGEST_HIGHLIGHT_STYLE="fg=#60C0FF,bold,underline"
 ZSH_AUTOSUGGEST_HISTORY_IGNORE="ls *,cd *"