Fix key listener bug