Minor change to allow keyboard interrupts