Fix CTRL+C behavior