Fix zsh function