diff --git a/common/test/acceptance/pages/lms/matlab_problem.py b/common/test/acceptance/pages/lms/matlab_problem.py
index c75b80fc57..ed02ca601d 100644
--- a/common/test/acceptance/pages/lms/matlab_problem.py
+++ b/common/test/acceptance/pages/lms/matlab_problem.py
@@ -39,5 +39,5 @@ class MatlabProblemPage(PageObject):
         """
         Returns the text value of given class.
         """
-        self.wait_for_ajax()
+        self.wait_for_element_visibility(class_name, 'Grader message is visible')
         return self.q(css=class_name).text