1 files changed, 8 insertions, 2 deletions
diff --git a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/model/Imports.java b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/model/Imports.java
index 9ed6097..c2e2a26 100644
--- a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/model/Imports.java
+++ b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/model/Imports.java
@@ -18,6 +18,12 @@ public class Imports {
     }
 
     public TypeRef add(TypeRef klass) {
+        // TODO: implement properly
+        if (true) {
+            imports.add(klass);
+            return klass;
+        }
+
         String fqName = klass.fqName;
         String name = klass.toString();
         for (TypeRef i : imports) {
@@ -32,7 +38,7 @@ public class Imports {
             }
         }
 
-        TypeRef ref = new TypeRef(fqName, name);
+        TypeRef ref = new TypeRef(fqName, name, klass.array, klass.args);
         imports.add(ref);
 
         return ref;
@@ -44,7 +50,7 @@ public class Imports {
             if (i.isPrimitive() || i.inUnnamedPackage()) {
                 continue;
             }
-            body.add("import " + i.fqName + ";");
+            body.add("import " + i.importName() + ";");
         }
         if (!imports.isEmpty()) {
             body.add("");