1 files changed, 7 insertions, 3 deletions
diff --git a/qa/qa/page/component/wiki_page_form.rb b/qa/qa/page/component/wiki_page_form.rb
index 7a7329e6110..9143a25d9ab 100644
--- a/qa/qa/page/component/wiki_page_form.rb
+++ b/qa/qa/page/component/wiki_page_form.rb
@@ -17,7 +17,10 @@ module QA
 
           base.view 'app/assets/javascripts/vue_shared/components/markdown/markdown_editor.vue' do
             element :markdown_editor_form_field
-            element :editing_mode_button
+          end
+
+          base.view 'app/assets/javascripts/vue_shared/components/markdown/editor_mode_dropdown.vue' do
+            element :editing_mode_switcher
           end
 
           base.view 'app/assets/javascripts/pages/shared/wikis/components/delete_wiki_modal.vue' do
@@ -55,8 +58,9 @@ module QA
         end
 
         def use_new_editor
-          within_element(:editing_mode_button) do
-            find('label', text: 'Rich text').click
+          click_element(:editing_mode_switcher)
+          within_element(:editing_mode_switcher) do
+            find('button', text: 'Rich text').click
           end
 
           wait_until(reload: false) do