diff --git a/core/tests/Drupal/Tests/Core/Entity/FieldDefinitionTest.php b/core/tests/Drupal/Tests/Core/Entity/FieldDefinitionTest.php
index 98fe41552c..e7c8d890c6 100644
--- a/core/tests/Drupal/Tests/Core/Entity/FieldDefinitionTest.php
+++ b/core/tests/Drupal/Tests/Core/Entity/FieldDefinitionTest.php
@@ -4,7 +4,7 @@
 
 use Drupal\Core\DependencyInjection\ContainerBuilder;
 use Drupal\Core\Entity\EntityFieldManagerInterface;
-use Drupal\Core\Field\BundleFieldDefinition;
+use Drupal\Core\Field\FieldDefinition;
 use Drupal\Core\Field\FieldStorageDefinitionInterface;
 use Drupal\Core\Field\FieldTypePluginManagerInterface;
 use Drupal\Core\TypedData\TypedDataManager;
@@ -363,25 +363,25 @@ public function factoryTypeProvider() {
    * @param string $factory_name
    *   The factory name to use.
    *
-   * @return \Drupal\Core\Field\BundleFieldDefinition
+   * @return \Drupal\Core\Field\FieldDefinition
    */
   protected function initializeFieldUsingFactory($factory_name) {
     switch ($factory_name) {
       case 'createFromFieldStorageDefinition':
-        return BundleFieldDefinition::createFromFieldStorageDefinition($this->storageDefinition);
+        return FieldDefinition::createFromFieldStorageDefinition($this->storageDefinition);
 
       case 'create':
-        $definition = BundleFieldDefinition::create($this->fieldType);
+        $definition = FieldDefinition::create($this->fieldType);
         $definition->setFieldStorageDefinition($this->storageDefinition);
         return $definition;
 
       case 'createFromDataType':
-        $definition = BundleFieldDefinition::createFromDataType($this->fieldType);
+        $definition = FieldDefinition::createFromDataType($this->fieldType);
         $definition->setFieldStorageDefinition($this->storageDefinition);
         return $definition;
 
       case 'createFromItemType':
-        $definition = BundleFieldDefinition::createFromItemType($this->fieldType);
+        $definition = FieldDefinition::createFromItemType($this->fieldType);
         $definition->setFieldStorageDefinition($this->storageDefinition);
         return $definition;
     }