Fix zsh support