Fix zsh setup script