I think that would be the appropriate thing to do. Change the code here to wait for the element to be visible as opposed to just present.