fix zsh vi mode prompt