Fix selenium tests with more checks

Animation causes elements to not be visible.
Wait for them to be displayed.