Fix zsh setup