Small fix after Ctrl+C implementation