diff --git a/ui/lib/core/addon/templates/components/alert-banner.hbs b/ui/lib/core/addon/templates/components/alert-banner.hbs
index 49c5e436d8ff8..1ccad621b0706 100644
--- a/ui/lib/core/addon/templates/components/alert-banner.hbs
+++ b/ui/lib/core/addon/templates/components/alert-banner.hbs
@@ -1,4 +1,4 @@
-<div data-test-flash-message="true">
+<div data-test-alert-banner>
   <div class="columns is-mobile is-variable is-1">
     <div class="column is-narrow message-icon">
       <Icon class={{this.alertType.glyphClass}} aria-hidden="true" @name={{this.alertType.glyph}} />
diff --git a/ui/tests/acceptance/client-history-test.js b/ui/tests/acceptance/client-history-test.js
index 066196846f1e8..feb5d518238fc 100644
--- a/ui/tests/acceptance/client-history-test.js
+++ b/ui/tests/acceptance/client-history-test.js
@@ -270,7 +270,7 @@ module('Acceptance | clients history tab', function (hooks) {
     await visit('/vault/clients/history');
     assert.equal(currentURL(), '/vault/clients/history', 'clients/history URL is correct');
     assert.dom(SELECTORS.activeTab).hasText('History', 'history tab is active');
-    assert.dom('[data-test-flash-message] .message-actions').containsText(`You upgraded to Vault 1.9.0`);
+    assert.dom('[data-test-alert-banner] .message-actions').containsText(`You upgraded to Vault 1.9.0`);
   });
 
   test('Shows empty if license start date is current month', async function (assert) {
diff --git a/ui/tests/acceptance/settings/mount-secret-backend-test.js b/ui/tests/acceptance/settings/mount-secret-backend-test.js
index f2a7f637070a3..f2e0e4d77bf25 100644
--- a/ui/tests/acceptance/settings/mount-secret-backend-test.js
+++ b/ui/tests/acceptance/settings/mount-secret-backend-test.js
@@ -89,7 +89,9 @@ module('Acceptance | settings/mount-secret-backend', function (hooks) {
     await page.enableEngine();
     await page.selectType('kv');
     await page.next().path(path).submit();
-    assert.dom('.alert-banner-message-body').containsText(`path is already in use at ${path}`);
+    assert
+      .dom('[data-test-alert-banner] .alert-banner-message-body')
+      .containsText(`path is already in use at ${path}`);
     assert.equal(currentRouteName(), 'vault.cluster.settings.mount-secret-backend');
 
     await page.secretList();