Fixes for MMS in Flame

This is a little bit boilerplate. getElement() performs a wait_for_element_displayed(), so maybe there's no need to do this