Fix namespace import for references

open-code-modeling · Dec 9, 2020 · dc356f7 · dc356f7
1 parent e564918
commit dc356f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ClassGenerator.php b/src/ClassGenerator.php
@@ -136,14 +136,14 @@ public function generateClasses(
                                     $propertyType->name()
                                 );
                                 $propertyClassName = ($this->classNameFilter)($propertyType->name());
-                                $classBuilder->addNamespaceImport($classNamespace . '\\' . $propertyClassName);
                             }
                             $classBuilder->addProperty(
                                 ClassPropertyBuilder::fromScratch(
                                     $propertyPropertyName,
                                     $propertyType->isNullable() ? ('?' . $propertyClassName) : $propertyClassName
                                 )
                             );
+                            $classBuilder->addNamespaceImport($classNamespace . '\\' . $propertyClassName);
                             break;
                         case $propertyType instanceof ScalarType:
                             $classBuilderCollection->add(