diff --git a/core/modules/migrate_drupal_ui/tests/src/Functional/MigrateUpgradeTestBase.php b/core/modules/migrate_drupal_ui/tests/src/Functional/MigrateUpgradeTestBase.php
index 6ff2d72..975388e 100644
--- a/core/modules/migrate_drupal_ui/tests/src/Functional/MigrateUpgradeTestBase.php
+++ b/core/modules/migrate_drupal_ui/tests/src/Functional/MigrateUpgradeTestBase.php
@@ -148,7 +148,10 @@ public function testMigrateUpgrade() {
     $this->drupalPostForm(NULL, $edits, t('Review upgrade'));
     $this->assertResponse(200);
     $this->assertText('Are you sure?');
+    $this->assertSession()->pageTextContains('Entities may be overwritten?');
+    $this->drupalPostForm(NULL, [], t('I acknowledge I may lose data, continue anyway.'));
     // Ensure we get errors about missing modules.
+    $this->assertText('Are you sure?');
     $this->assertSession()->pageTextContains(t('Source module not found for migration_provider_no_annotation.'));
     $this->assertSession()->pageTextContains(t('Source module not found for migration_provider_test.'));
     $this->assertSession()->pageTextContains(t('Destination module not found for migration_provider_test'));
@@ -167,8 +170,10 @@ public function testMigrateUpgrade() {
     $this->drupalPostForm(NULL, $edits, t('Review upgrade'));
     $this->assertSession()->statusCodeEquals(200);
     $this->assertSession()->pageTextContains('Are you sure?');
+    $this->assertSession()->pageTextContains('Entities may be overwritten?');
     $this->drupalPostForm(NULL, [], t('I acknowledge I may lose data, continue anyway.'));
     // Ensure there are no errors about the missing modules from the test module.
+    $this->assertText('Are you sure?');
     $this->assertSession()->pageTextNotContains(t('Source module not found for migration_provider_no_annotation.'));
     $this->assertSession()->pageTextNotContains(t('Source module not found for migration_provider_test.'));
     $this->assertSession()->pageTextNotContains(t('Destination module not found for migration_provider_test'));