diff --git a/core/modules/migrate/src/Tests/MigrationGroupTest.php b/core/modules/migrate/src/Tests/MigrationGroupTest.php
index 997b4ff..25ea61e 100644
--- a/core/modules/migrate/src/Tests/MigrationGroupTest.php
+++ b/core/modules/migrate/src/Tests/MigrationGroupTest.php
@@ -24,7 +24,6 @@ class MigrationGroupTest extends WebTestBase {
    * {@inheritdoc}
    */
   protected function setUp() {
-    $this->strictConfigSchema = FALSE;
     parent::setUp();
   }
 
@@ -45,16 +44,20 @@ public function testConfigurationMerge() {
           'cardinality' => '1', // In migration, so will be overridden.
         ),
       ),
-      'destination' => array('entity' => 'field_storage_config'), // Not in migration, so will be added.
+      'destination' => array('plugin' => 'field_storage_config'), // Not in migration, so will be added.
     ));
     $migration_group->save();
 
     /** @var MigrationInterface $migration */
-    $migration = entity_create('migration', array());
-    $migration->set('id', 'specific_migration');
-    $migration->set('migration_group', $group_id);
-    $migration->set('label', 'Unaffected by the group');
-    $migration->set('migration_groups', array('Drupal 7')); // Overrides group.
+    $migration = entity_create('migration', array(
+      'id' => 'specific_migration',
+      'load' => [],
+      'migration_group' => $group_id,
+      'label' => 'Unaffected by the group',
+      'migration_groups' => array('Drupal 7'), // Overrides group.
+      'destination' => array(),
+      'source' => array(),
+    ));
     $migration->set('source', array(
       'plugin' => 'empty',        // Not in group, persists.
       'constants' => array(
@@ -76,7 +79,7 @@ public function testConfigurationMerge() {
           'cardinality' => '3',
         ),
       ),
-      'destination' => array('entity' => 'field_storage_config'),
+      'destination' => array('plugin' => 'field_storage_config'),
     );
     /** @var MigrationInterface $loaded_migration */
     $loaded_migration = entity_load('migration', 'specific_migration', TRUE);
@@ -96,9 +99,14 @@ public function testDelete() {
     $migration_group->save();
 
     /** @var MigrationInterface $migration */
-    $migration = entity_create('migration', array());
-    $migration->set('id', 'specific_migration');
-    $migration->set('migration_group', 'test_group');
+    $migration = entity_create('migration', [
+      'id' => 'specific_migration',
+      'migration_group' => 'test_group',
+      'migration_groups' => array(),
+      'load' => [],
+      'destination' => array(),
+      'source' => array(),
+    ]);
     $migration->save();
 
     /** @var MigrationGroupInterface $loaded_migration_group */