1 files changed, 4 insertions, 1 deletions
diff --git a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/model/ClassG.java b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/model/ClassG.java
index 6bfd0be..9befa36 100644
--- a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/model/ClassG.java
+++ b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/model/ClassG.java
@@ -167,7 +167,10 @@ public class ClassG {
             parameters.add("final " + p.klass.name + " " + p.name);
         }
 
-        writer.print("    public " + returnString + " " + method.name + "(" + collectionToDelimitedString(parameters, ", ") + ")");
+        writer.print("    " +
+                Modifier.toString(method.modifiers) + " " +
+                returnString + " " +
+                method.name + "(" + collectionToDelimitedString(parameters, ", ") + ")");
         if (method.exceptions.isEmpty()) {
             writer.println(" {");
         } else {