Tweak jez.zsh-theme to work with/without vi-mode.