We should catch KeyboardInterrupt in __init__.py

We don't catch it if someone install it via setup.py with the old fix.