Fix exit on return key