It might be worth adding `wait_for_visible` and `wait_for_invisible` helpers here to make this step a little more robust.