Another fix on ZSH prompt