Multiple test fixes

Not needed. getElement internally performs a wait_for_element_