diff --git a/tests/selenium/pageobjects/viewedit.page.js b/tests/selenium/pageobjects/viewedit.page.js
index e8ef7b727..02ded7fc8 100644
--- a/tests/selenium/pageobjects/viewedit.page.js
+++ b/tests/selenium/pageobjects/viewedit.page.js
@@ -50,11 +50,11 @@ class ViewEditPage extends Page {
 	}
 
 	get error() {
-		return $( '.mw-message-box-error' );
+		return $( '.cdx-message--error' );
 	}
 
 	get warning() {
-		return $( '.mw-message-box-warning' );
+		return $( '.cdx-message--warning' );
 	}
 
 	async submit() {
diff --git a/tests/selenium/pageobjects/viewlist.page.js b/tests/selenium/pageobjects/viewlist.page.js
index 918bb3e93..7117b908f 100644
--- a/tests/selenium/pageobjects/viewlist.page.js
+++ b/tests/selenium/pageobjects/viewlist.page.js
@@ -12,7 +12,7 @@ class ViewListPage extends Page {
 	}
 
 	get filterSavedNotice() {
-		return $( '.mw-message-box-success' );
+		return $( '.cdx-message--success' );
 	}
 
 	async savedFilterID() {