diff --git a/mod/assign/tests/behat/grading_status.feature b/mod/assign/tests/behat/grading_status.feature
index f265f3ea8d2..410cfab554a 100644
--- a/mod/assign/tests/behat/grading_status.feature
+++ b/mod/assign/tests/behat/grading_status.feature
@@ -101,8 +101,8 @@ Feature: View the grading status of an assignment
     And I press "Continue"
     And I should see "In marking" in the "Student 1" "table_row"
     # The grade should also remain displayed as it's stored in the assign DB tables, but the final grade should be empty.
-    And I should see "50.00" in the "#mod_assign_grading_r0_c5" "css_element"
-    And I should see "-" in the "#mod_assign_grading_r0_c13" "css_element"
+    And "Student 1" row "Grade" column of "generaltable" table should contain "50.00"
+    And "Student 1" row "Final grade" column of "generaltable" table should contain "-"
     And I log out
 
   @javascript