Fix zsh completion

Fixes: #520