Fix zsh completion