Fix KeyboardInterrupt exception commented out