Fix the zsh prompt