diff --git a/modules/ve-mw/tests/browser/features/support/pages/visual_editor_page.rb b/modules/ve-mw/tests/browser/features/support/pages/visual_editor_page.rb
index 161f3ccfe7..5470d80cc2 100644
--- a/modules/ve-mw/tests/browser/features/support/pages/visual_editor_page.rb
+++ b/modules/ve-mw/tests/browser/features/support/pages/visual_editor_page.rb
@@ -72,7 +72,7 @@ class VisualEditorPage
   div(:link_overlay_does_not_exist, text: /New page/)
   div(:link_overlay_external_link, text: /External link.+New page/m)
   div(:link_overlay_wiki_page, text: /Matching page/)
-  unordered_list(:media_caption, class: 've-ui-contextMenuWidget')
+  unordered_list(:media_caption, class: 've-ui-contextSelectWidget')
   span(:media_exit, text: 'Return to save form')
   figure(:media_image, index: 0)
   span(:media_insert_menu, class: 'oo-ui-tool-name-media')
@@ -98,7 +98,7 @@ class VisualEditorPage
   a(:page_title, text: /Page title/)
   a(:paragraph, text: /Paragraph/)
   div(:parameter_icon, text: 'q')
-  unordered_list(:popup_icon, class: 've-ui-contextMenuWidget')
+  unordered_list(:popup_icon, class: 've-ui-contextSelectWidget')
   a(:preformatted, text: /Preformatted/)
   label(:prevent_redirect, class: 'oo-ui-layout oo-ui-labelElement oo-ui-fieldLayout oo-ui-fieldLayout-align-inline oo-ui-fieldLayout-disabled')
   span(:ref_list_insert_menu, class: 'oo-ui-tool-name-referencesList')