1 files changed, 6 insertions, 6 deletions
diff --git a/spec/features/admin/admin_sees_background_migrations_spec.rb b/spec/features/admin/admin_sees_background_migrations_spec.rb
index 4b8636da6b4..cad1bf74d2e 100644
--- a/spec/features/admin/admin_sees_background_migrations_spec.rb
+++ b/spec/features/admin/admin_sees_background_migrations_spec.rb
@@ -92,11 +92,11 @@ RSpec.describe "Admin > Admin sees background migrations", feature_category: :da
       expect(page).not_to have_content('Paused')
       expect(page).to have_content('Active')
 
-      click_button('Pause')
+      click_on('Pause')
       expect(page).not_to have_content('Active')
       expect(page).to have_content('Paused')
 
-      click_button('Resume')
+      click_on('Resume')
       expect(page).not_to have_content('Paused')
       expect(page).to have_content('Active')
     end
@@ -123,7 +123,7 @@ RSpec.describe "Admin > Admin sees background migrations", feature_category: :da
         tab = find_link 'Failed'
         tab.click
 
-        expect(page).to have_selector("[method='post'][action='/admin/background_migrations/#{failed_migration.id}/retry?database=main']")
+        expect(page).to have_selector("[data-method='post'][href='/admin/background_migrations/#{failed_migration.id}/retry?database=main']")
       end
     end
 
@@ -144,7 +144,7 @@ RSpec.describe "Admin > Admin sees background migrations", feature_category: :da
         expect(page).to have_content('0.00%')
         expect(page).to have_content(failed_migration.status_name.to_s)
 
-        click_button('Retry')
+        click_on('Retry')
         expect(page).not_to have_content(failed_migration.job_class_name)
         expect(page).not_to have_content(failed_migration.table_name)
         expect(page).not_to have_content('0.00%')
@@ -172,7 +172,7 @@ RSpec.describe "Admin > Admin sees background migrations", feature_category: :da
   end
 
   it 'can change tabs and retain database param' do
-    skip_if_multiple_databases_not_setup
+    skip_if_multiple_databases_not_setup(:ci)
 
     visit admin_background_migrations_path(database: 'ci')
 
@@ -212,7 +212,7 @@ RSpec.describe "Admin > Admin sees background migrations", feature_category: :da
 
     context 'when multi database is enabled' do
       before do
-        skip_if_multiple_databases_not_setup
+        skip_if_multiple_databases_not_setup(:ci)
 
         allow(Gitlab::Database).to receive(:db_config_names).and_return(%w[main ci])
       end