1 files changed, 2 insertions, 1 deletions
diff --git a/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java b/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java
index 9be591c..b41c972 100644
--- a/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java
+++ b/src/main/java/org/zwobble/couscous/ast/ClassNodeBuilder.java
@@ -1,6 +1,7 @@
 package org.zwobble.couscous.ast;
 
 import com.google.common.collect.ImmutableList;
+import org.zwobble.couscous.ast.identifiers.Identifier;
 
 import java.util.Optional;
 import java.util.function.Function;
@@ -110,7 +111,7 @@ public class ClassNodeBuilder {
             return ThisReferenceNode.thisReference(name);
         }
         
-        public MethodBuilder<T> argument(String id, String name, TypeName type) {
+        public MethodBuilder<T> argument(Identifier id, String name, TypeName type) {
             arguments.add(formalArg(var(id, name, type)));
             return this;
         }