Fix ZSH setup