Fixing 'enter' bug