fix on previous commit

raffaeler · Oct 27, 2021 · 7c3c3f5 · 7c3c3f5
1 parent cdc9b55
commit 7c3c3f5
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/SharedGenerators/GeneratorManager.cs b/src/SharedGenerators/GeneratorManager.cs
@@ -44,10 +44,13 @@ public SourceText GenerateINPCClass(string? @namespace, string className,
                 if (field == null || field.AttributeArguments == null ||
                     field.FieldType == null || field.FieldName == null) continue;
 
-                if (field.FieldTypeNamespace != null
-                    && field.FieldTypeNamespace != field.NamespaceName)
+                if (field.FieldTypeNamespaces != null)
                 {
-                    gen.Usings.Add(field.FieldTypeNamespace);
+                    if(field.NamespaceName != null)
+                        field.FieldTypeNamespaces.Remove(field.NamespaceName);
+
+                    foreach(var ns in field.FieldTypeNamespaces)
+                        gen.Usings.Add(ns);
                 }
 
                 var partialMethod = field.AttributeArguments.ExtraNotify