Fix auto disable when exiting insert mode.

Hi

I noticed that the automatic disable when exiting insert mode didn't work in the latest version.
So I have attempted to fix that.

This was the leas intrusive way I could find.
Though I'm not sure what your plan for the `'i'` and `'c'` arguments are. So maybe it is a mistake to remove those.

Thank you for this great plugin btw.