diff --git a/core/modules/migrate_drupal_ui/src/Form/MigrateUpgradeForm.php b/core/modules/migrate_drupal_ui/src/Form/MigrateUpgradeForm.php
index e998cee..06cfa18 100644
--- a/core/modules/migrate_drupal_ui/src/Form/MigrateUpgradeForm.php
+++ b/core/modules/migrate_drupal_ui/src/Form/MigrateUpgradeForm.php
@@ -407,7 +407,7 @@ public function buildConfirmForm(array $form, FormStateInterface $form_state) {
       $source_module = NULL;
       $source_plugin = $migration->getSourcePlugin();
       if ($source_plugin instanceof MigrationProviderInterface) {
-        $source_module = $source_plugin->getProvider();
+        $source_module = $source_plugin->getMigrationProvider();
       }
       if (!$source_module) {
         drupal_set_message($this->t('Source module not found for @migration_id.', ['@migration_id' => $migration_id]), 'error');
@@ -416,7 +416,7 @@ public function buildConfirmForm(array $form, FormStateInterface $form_state) {
       $destination_module = NULL;
       $destination_plugin = $migration->getDestinationPlugin();
       if ($destination_plugin instanceof MigrationProviderInterface) {
-        $destination_module = $destination_plugin->getProvider();
+        $destination_module = $destination_plugin->getMigrationProvider();
       }
       if (!$destination_module) {
         drupal_set_message($this->t('Destination module not found for @migration_id.', ['@migration_id' => $migration_id]), 'error');