1 files changed, 3 insertions, 2 deletions
diff --git a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/TransactionalHandler.java b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/TransactionalHandler.java
index b5db075..bfa1ec8 100644
--- a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/TransactionalHandler.java
+++ b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/TransactionalHandler.java
@@ -1,6 +1,7 @@
 package io.trygvis.container.compiler;
 
 import io.trygvis.container.compiler.model.ClassG;
+import io.trygvis.container.compiler.model.Constructor;
 import io.trygvis.container.compiler.model.FieldRef;
 import io.trygvis.container.compiler.model.MethodRef;
 import io.trygvis.container.compiler.model.Parameters;
@@ -109,7 +110,7 @@ public class TransactionalHandler extends AbstractHandler {
             body.add("        });");
 
             MethodRef m = new MethodRef(PUBLIC, returnType, ee.getSimpleName().toString(), parameters, body);
-            g.addMethod(m);
+            g.add(m);
         }
 
         JavaFileObject sourceFile = processingEnv.getFiler().createSourceFile((p.length() == 0 ? "" : p + ".") + className, element);
@@ -137,6 +138,6 @@ public class TransactionalHandler extends AbstractHandler {
         List<String> body = new ArrayList<>();
         body.add("super(" + collectionToDelimitedString(goesToSuper, ", ") + ");");
         body.add("this." + platformTransactionManager.name + " = " + transactionManager.name + ";");
-        g.addConstructor(parameters, body);
+        g.add(new Constructor(parameters, body));
     }
 }