I would consider adding this new wait before the existing wait, rather than replacing the existing wait, as I'm concerned about not waiting for an element that we need to interact with. Also the previous wait was a ```wait_for_element_present```, but should really be ```wait_for_element_displayed```. I wonder if just changing it to that would also fix the problem?