diff --git a/common/test/acceptance/pages/lms/edxnotes.py b/common/test/acceptance/pages/lms/edxnotes.py index 67bde2ec96..2ac9599a50 100644 --- a/common/test/acceptance/pages/lms/edxnotes.py +++ b/common/test/acceptance/pages/lms/edxnotes.py @@ -564,7 +564,7 @@ class EdxNoteHighlight(NoteChild): """ Clicks cancel button. """ - self.q(css=self._bounded_selector(".annotator-cancel")).first.click() + self.q(css=self._bounded_selector(".annotator-close")).first.click() self.wait_for_notes_invisibility("Note is canceled.") return self @@ -605,8 +605,7 @@ class EdxNoteHighlight(NoteChild): text = element.text[0].strip() else: text = None - self.q(css=("body")).first.click() - self.wait_for_notes_invisibility() + self.cancel() return text @text.setter @@ -629,8 +628,7 @@ class EdxNoteHighlight(NoteChild): if tags: for tag in tags: tag_text.append(tag.text) - self.q(css="body").first.click() - self.wait_for_notes_invisibility() + self.cancel() return tag_text @tags.setter