why not just use `on_focus`?