diff --git a/src/Builder/Capacity/Extractor/All.php b/src/Builder/Capacity/Extractor/All.php
index 0e1eb8a..1fa6c0a 100644
--- a/src/Builder/Capacity/Extractor/All.php
+++ b/src/Builder/Capacity/Extractor/All.php
@@ -134,7 +134,7 @@ private function compileArguments(): array
             );
         }
 
-        if (null !== $this->referenceEntity) {
+        if (null !== $this->referenceEntityAttributeCode) {
             $args[] = new Node\Arg(
                 value: $this->referenceEntityAttributeCode,
                 name: new Node\Identifier('attributeCode'),