Set keyboard input listener to document.body again