diff --git a/mod/assign/feedback/editpdf/tests/behat/annotate_pdf.feature b/mod/assign/feedback/editpdf/tests/behat/annotate_pdf.feature
index 10e533ce4b9..a5e269abe36 100644
--- a/mod/assign/feedback/editpdf/tests/behat/annotate_pdf.feature
+++ b/mod/assign/feedback/editpdf/tests/behat/annotate_pdf.feature
@@ -58,7 +58,7 @@ Feature: In an assignment, teacher can annotate PDF files during grading
     And I click on ".stampbutton" "css_element"
     And I click on ".linebutton" "css_element"
     And I click on ".commentcolourbutton" "css_element"
-    And I click on "//img[@alt=\"Blue\"]" "xpath_element"
+    And I click on "//img[@alt=\"Blue\"]/parent::button" "xpath_element"
     And I change window size to "medium"
     And I wait until the page is ready
     And I click on "Close" "button"