if you change this line to:
`done = self.wait_for_element_present(*self._done_button_locator)`

then you don't need any of the extra code you added.