Fixed the KeyListener