Fixed bug where keyboard couldn't be used after the context menu is closed