Can you use `focusElementByQuery()` here?