Bug fix - return filtered staged builders

Randgalt · Jun 7, 2024 · 7b29c3b · 7b29c3b
1 parent df0f722
commit 7b29c3b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...ssor/src/main/java/io/soabase/recordbuilder/processor/InternalRecordBuilderProcessor.java b/...ssor/src/main/java/io/soabase/recordbuilder/processor/InternalRecordBuilderProcessor.java
@@ -787,7 +787,8 @@ private void addStaticStagedBuilderMethod(String builderMethodName) {
         codeBlock.addStatement("() -> $L", uniqueVarName);
         IntStream.range(0, filteredRecordComponents.size()).forEach(__ -> codeBlock.unindent().addStatement("}"));
 
-        var returnType = stagedBuilderType(recordComponents.isEmpty() ? builderClassType : recordComponents.get(0));
+        var returnType = stagedBuilderType(
+                filteredRecordComponents.isEmpty() ? builderClassType : filteredRecordComponents.get(0));
 
         var methodSpec = MethodSpec.methodBuilder(builderMethodName)
                 .addJavadoc("Return the first stage of a staged builder\n")