diff --git a/tests/src/Functional/DynamicEntityReferenceFieldDefaultValueTest.php b/tests/src/Functional/DynamicEntityReferenceFieldDefaultValueTest.php
index e69fd63549274ad4bf7d29f4751b96b3916d31a7..a3b0cd5436a68fa303e751c29441fb6bd3f773a1 100644
--- a/tests/src/Functional/DynamicEntityReferenceFieldDefaultValueTest.php
+++ b/tests/src/Functional/DynamicEntityReferenceFieldDefaultValueTest.php
@@ -2,11 +2,11 @@
 
 namespace Drupal\Tests\dynamic_entity_reference\Functional;
 
-use Drupal\config\Tests\SchemaCheckTestTrait;
 use Drupal\field\Entity\FieldConfig;
 use Drupal\field\Entity\FieldStorageConfig;
 use Drupal\node\Entity\Node;
 use Drupal\Tests\BrowserTestBase;
+use Drupal\Tests\SchemaCheckTestTrait;
 
 /**
  * Tests dynamic entity reference field default values storage in CMI.
diff --git a/tests/src/Kernel/DynamicEntityReferenceBaseFieldRevisionTest.php b/tests/src/Kernel/DynamicEntityReferenceBaseFieldRevisionTest.php
index cd61a766881e8b7add25078a2ee6264c6bf9c2d9..dba7e499e9ad5ee72027e72213f0dcca3325b1c5 100644
--- a/tests/src/Kernel/DynamicEntityReferenceBaseFieldRevisionTest.php
+++ b/tests/src/Kernel/DynamicEntityReferenceBaseFieldRevisionTest.php
@@ -2,9 +2,9 @@
 
 namespace Drupal\Tests\dynamic_entity_reference\Kernel;
 
-use Drupal\config\Tests\SchemaCheckTestTrait;
 use Drupal\Core\Field\FieldStorageDefinitionInterface;
 use Drupal\KernelTests\Core\Entity\EntityKernelTestBase;
+use Drupal\Tests\SchemaCheckTestTrait;
 
 /**
  * Tests for the dynamic entity reference base field for revisionable entities.
diff --git a/tests/src/Kernel/DynamicEntityReferenceBaseFieldTest.php b/tests/src/Kernel/DynamicEntityReferenceBaseFieldTest.php
index 2e76ca51dcc4b52682b82832762b3a896b34df12..31d5a027ac2d3cb8b2263a40d1729c90fd540ec3 100644
--- a/tests/src/Kernel/DynamicEntityReferenceBaseFieldTest.php
+++ b/tests/src/Kernel/DynamicEntityReferenceBaseFieldTest.php
@@ -2,11 +2,11 @@
 
 namespace Drupal\Tests\dynamic_entity_reference\Kernel;
 
-use Drupal\config\Tests\SchemaCheckTestTrait;
 use Drupal\Core\Field\FieldStorageDefinitionInterface;
 use Drupal\entity_test\Entity\EntityTest;
 use Drupal\entity_test\Entity\EntityTestMul;
 use Drupal\KernelTests\Core\Entity\EntityKernelTestBase;
+use Drupal\Tests\SchemaCheckTestTrait;
 
 /**
  * Tests for the dynamic entity reference base field.
diff --git a/tests/src/Kernel/DynamicEntityReferenceFieldTest.php b/tests/src/Kernel/DynamicEntityReferenceFieldTest.php
index e728d31d026359b9c5438eacbca1798f744d9548..65953fe5bce00321992485ef6d1c41bc675f016d 100644
--- a/tests/src/Kernel/DynamicEntityReferenceFieldTest.php
+++ b/tests/src/Kernel/DynamicEntityReferenceFieldTest.php
@@ -2,12 +2,12 @@
 
 namespace Drupal\Tests\dynamic_entity_reference\Kernel;
 
-use Drupal\config\Tests\SchemaCheckTestTrait;
 use Drupal\Core\Field\FieldStorageDefinitionInterface;
 use Drupal\entity_test\Entity\EntityTestBundle;
 use Drupal\field\Entity\FieldConfig;
 use Drupal\field\Entity\FieldStorageConfig;
 use Drupal\KernelTests\Core\Entity\EntityKernelTestBase;
+use Drupal\Tests\SchemaCheckTestTrait;
 
 /**
  * Tests for the dynamic entity reference field.
diff --git a/tests/src/Kernel/DynamicEntityReferenceSchemaTest.php b/tests/src/Kernel/DynamicEntityReferenceSchemaTest.php
index 07e66472067c51338d75e56e6ac7bbd71f10daf1..dbc1d4a8afe1caa4340bbb0f31591579a88e6b01 100644
--- a/tests/src/Kernel/DynamicEntityReferenceSchemaTest.php
+++ b/tests/src/Kernel/DynamicEntityReferenceSchemaTest.php
@@ -2,13 +2,13 @@
 
 namespace Drupal\Tests\dynamic_entity_reference\Kernel;
 
-use Drupal\config\Tests\SchemaCheckTestTrait;
 use Drupal\Core\Entity\Entity\EntityFormDisplay;
 use Drupal\Core\Entity\Entity\EntityViewDisplay;
 use Drupal\Core\Field\FieldStorageDefinitionInterface;
 use Drupal\field\Entity\FieldConfig;
 use Drupal\field\Entity\FieldStorageConfig;
 use Drupal\KernelTests\Core\Entity\EntityKernelTestBase;
+use Drupal\Tests\SchemaCheckTestTrait;
 
 /**
  * Ensures that Dynamic Entity References schema is correct.