diff --git a/generator/src/main/java/org/sudu/protogen/generator/message/MessageBuilderGenerator.java b/generator/src/main/java/org/sudu/protogen/generator/message/MessageBuilderGenerator.java
index ef45393..d2fe1d8 100644
--- a/generator/src/main/java/org/sudu/protogen/generator/message/MessageBuilderGenerator.java
+++ b/generator/src/main/java/org/sudu/protogen/generator/message/MessageBuilderGenerator.java
@@ -65,6 +65,7 @@ private MethodSpec setterForField(FieldSpec fieldSpec, String builderName) {
         return MethodSpec.methodBuilder("set" + Name.toCamelCase(fieldSpec.name))
                 .addModifiers(Modifier.PUBLIC)
                 .returns(ClassName.get("",builderName))
+                .addParameter(Poem.fieldToParameter(fieldSpec))
                 .addStatement("this.$N = $N", fieldSpec, fieldSpec)
                 .addStatement("return this")
                 .build();