ZSH fix 

This fixes the issue with zsh interactive mode.