Fix setup for zsh