It seems like adding `self.marionette.switch_to_frame(self.apps.displayed_app.frame)` here fixes the issue.