diff --git a/tests/src/Kernel/MigrationGroupTest.php b/tests/src/Kernel/MigrationGroupTest.php
index 53d3ad467d52104f626c74b256e539a0c4a5d087..2909891c644be3e91b1fd921f053758da7a54ff2 100644
--- a/tests/src/Kernel/MigrationGroupTest.php
+++ b/tests/src/Kernel/MigrationGroupTest.php
@@ -69,10 +69,9 @@ class MigrationGroupTest extends KernelTestBase {
     $migration->save();
 
     $expected_config = [
-      'migration_group' => $group_id,
       'label' => 'Unaffected by the group',
-      'migration_tags' => ['Drupal 7'],
-      'source' => [
+      'getMigrationTags' => ['Drupal 7'],
+      'getSourceConfiguration' => [
         'plugin' => 'empty',
         'constants' => [
           'entity_type' => 'user',
@@ -80,13 +79,13 @@ class MigrationGroupTest extends KernelTestBase {
           'cardinality' => '3',
         ],
       ],
-      'destination' => ['plugin' => 'field_storage_config'],
+      'getDestinationConfiguration' => ['plugin' => 'field_storage_config'],
     ];
-    /** @var \Drupal\migrate\Plugin\MigrationInterface $loaded_migration */
+    /** @var \Drupal\migrate_plus\Plugin\MigrationInterface $loaded_migration */
     $loaded_migration = $this->container->get('plugin.manager.migration')
       ->createInstance('specific_migration');
-    foreach ($expected_config as $key => $expected_value) {
-      $actual_value = $loaded_migration->get($key);
+    foreach ($expected_config as $method => $expected_value) {
+      $actual_value = call_user_func([$loaded_migration, $method]);
       $this->assertEquals($expected_value, $actual_value);
     }
   }
diff --git a/tests/src/Kernel/Plugin/migrate/process/EntityGenerateTest.php b/tests/src/Kernel/Plugin/migrate/process/EntityGenerateTest.php
index 0240c1049df2efd68df2f5bdcc3e2082eee53ccf..fb9f7891507fb5688e7f1937a2e1f6804328ad9e 100644
--- a/tests/src/Kernel/Plugin/migrate/process/EntityGenerateTest.php
+++ b/tests/src/Kernel/Plugin/migrate/process/EntityGenerateTest.php
@@ -7,13 +7,13 @@ use Drupal\Core\Field\FieldStorageDefinitionInterface;
 use Drupal\Core\Language\LanguageInterface;
 use Drupal\field\Entity\FieldConfig;
 use Drupal\field\Entity\FieldStorageConfig;
-use Drupal\field\Tests\EntityReference\EntityReferenceTestTrait;
 use Drupal\KernelTests\KernelTestBase;
 use Drupal\migrate\MigrateExecutable;
 use Drupal\migrate\MigrateMessageInterface;
 use Drupal\node\Entity\NodeType;
 use Drupal\taxonomy\Entity\Term;
 use Drupal\taxonomy\Entity\Vocabulary;
+use Drupal\Tests\field\Traits\EntityReferenceTestTrait;
 
 /**
  * Tests the migration plugin.