diff --git a/core/modules/system/lib/Drupal/system/Tests/Entity/EntityFieldTest.php b/core/modules/system/lib/Drupal/system/Tests/Entity/EntityFieldTest.php
index 1e777a6..4145c85 100644
--- a/core/modules/system/lib/Drupal/system/Tests/Entity/EntityFieldTest.php
+++ b/core/modules/system/lib/Drupal/system/Tests/Entity/EntityFieldTest.php
@@ -539,8 +539,8 @@ public function testEntityConstraintValidation() {
     $definition = FieldDefinition::create('entity_reference')
       ->setLabel('Test entity')
       ->setSetting('target_type', 'entity_test');
-    $reference_field_item = \Drupal::TypedData()->create($definition);
-    $reference = $reference_field_item->get('entity');
+    $reference_field = \Drupal::TypedData()->create($definition);
+    $reference = $reference_field->first()->get('entity');
     $reference->setValue($entity);
 
     // Test validation the typed data object.
@@ -565,8 +565,8 @@ public function testEntityConstraintValidation() {
         'target_type' => 'node',
         'target_bundle' => 'article',
       ));
-    $reference_field_item = \Drupal::TypedData()->create($definition);
-    $reference = $reference_field_item->get('entity');
+    $reference_field = \Drupal::TypedData()->create($definition);
+    $reference = $reference_field->first()->get('entity');
     $reference->setValue($node);
     $violations = $reference->validate();
     $this->assertEqual($violations->count(), 1);