Selectmenu: make sure ui-state-focus class is set on button after initial focus