Fix ctrl+c issue