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 ae47095..96cf361 100644
--- a/core/modules/migrate_drupal_ui/tests/src/Functional/MigrateUpgradeTestBase.php
+++ b/core/modules/migrate_drupal_ui/tests/src/Functional/MigrateUpgradeTestBase.php
@@ -183,15 +183,15 @@ public function testMigrateUpgrade() {
     $all_available = $this->getAvailablePaths();
     $session = $this->assertSession();
     foreach ($all_available as $available) {
-      $session->elementExists('xpath', '//table[@id="edit-available-module-list-module-list"]//td[text()="' . $available . '"]');
-      $session->elementNotExists('xpath', '//table[@id="edit-missing-module-list"]//td[text()="' . $available . '"]');
+      $session->elementExists('xpath', "//span[contains(@class, 'checked') and text() = '$available']");
+      $session->elementNotExists('xpath', "//span[contains(@class, 'warning') and text() = '$available']");
     }
 
     // Test the missing migration paths.
     $all_missing = $this->getMissingPaths();
     foreach ($all_missing as $missing) {
-      $session->elementExists('xpath', '//table[@id="edit-missing-module-list"]//td[text()="' . $missing . '"]');
-      $session->elementNotExists('xpath', '//table[@id="edit-available-module-list-module-list"]//td[text()="' . $missing . '"]');
+      $session->elementExists('xpath', "//span[contains(@class, 'warning') and text() = '$missing']");
+      $session->elementNotExists('xpath', "//span[contains(@class, 'checked') and text() = '$missing']");
     }
 
     $this->drupalPostForm(NULL, [], t('Perform upgrade'));