diff --git a/tests/src/Kernel/EntityQueryRelationshipTest.php b/tests/src/Kernel/EntityQueryRelationshipTest.php
index c82edbe..5de7ee9 100644
--- a/tests/src/Kernel/EntityQueryRelationshipTest.php
+++ b/tests/src/Kernel/EntityQueryRelationshipTest.php
@@ -132,8 +132,8 @@ class EntityQueryRelationshipTest extends EntityKernelTestBase {
     // This returns the 0th entity as that's only one pointing to the 0th
     // account.
     $this->queryResults = $this->factory->get('entity_test')
-      ->condition("field_test.0.target_id.entity:entity_test_rev.name", 'Foobar')
-      ->condition("field_test.1.target_id.entity:entity_test_rev.name", 'Barfoo')
+      ->condition("field_test.0.entity:entity_test_rev.name", 'Foobar')
+      ->condition("field_test.1.entity:entity_test_rev.name", 'Barfoo')
       ->execute();
     $this->assertEquals([1 => 1, 2 => 2], $this->queryResults);
   }