diff --git a/vizro-ai/tests/vizro_ai_ui/test_vizro_ai_ui.py b/vizro-ai/tests/vizro_ai_ui/test_vizro_ai_ui.py
index d5d55423b..83f53056f 100644
--- a/vizro-ai/tests/vizro_ai_ui/test_vizro_ai_ui.py
+++ b/vizro-ai/tests/vizro_ai_ui/test_vizro_ai_ui.py
@@ -50,7 +50,7 @@ def _text_waiter():
         try:
             webdriver_waiter(
                 chromedriver,
-                '//*[starts-with(@class, "language-python")]',
+                '//*[starts-with(@class, "language-pytho")]',
             )
             return True
         except (TimeoutException, InvalidSelectorException):