Handle KeyboardInterrupt exceptions correctly