diff --git a/core/modules/system/tests/modules/entity_test/entity_test.install b/core/modules/system/tests/modules/entity_test/entity_test.install
index ffefe022f9..69510aa9cc 100644
--- a/core/modules/system/tests/modules/entity_test/entity_test.install
+++ b/core/modules/system/tests/modules/entity_test/entity_test.install
@@ -5,9 +5,9 @@
  * Install, update and uninstall functions for the entity_test module.
  */
 
-use Drupal\system\Tests\Update\DbUpdatesTrait;
 use Drupal\field\Entity\FieldStorageConfig;
 use Drupal\field\Entity\FieldConfig;
+use Drupal\Tests\system\Functional\Update\DbUpdatesTrait;
 
 /**
  * Implements hook_install().
diff --git a/core/modules/system/tests/modules/entity_test_update/entity_test_update.install b/core/modules/system/tests/modules/entity_test_update/entity_test_update.install
index 1423308500..3ac7a85692 100644
--- a/core/modules/system/tests/modules/entity_test_update/entity_test_update.install
+++ b/core/modules/system/tests/modules/entity_test_update/entity_test_update.install
@@ -5,6 +5,6 @@
  * Install, update and uninstall functions for the Entity Test Update module.
  */
 
-use Drupal\system\Tests\Update\DbUpdatesTrait;
+use Drupal\Tests\system\Functional\Update\DbUpdatesTrait;
 
 DbUpdatesTrait::includeUpdates('entity_test_update', 'entity_rev_pub_updates');