Fixing KeyListener.