display: Fix checks for KeyPress/ButtonPress