[zsh] Fix the prompt