1 files changed, 9 insertions, 5 deletions
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatypes/JavaTypesGenerator.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatypes/JavaTypesGenerator.java
index 12c1451..0526ca8 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatypes/JavaTypesGenerator.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatypes/JavaTypesGenerator.java
@@ -260,11 +260,7 @@ public class JavaTypesGenerator implements Generator {
                 ))
             ));
 
-            Optional<JavaTypeRef> fieldBuilderTypeRef = switch (field.type().value()) {
-                case StructType structType -> Optional.of(generateBuilderTypeRef(structType));
-                case SumType sumType -> Optional.of(generateBuilderTypeRef(sumType));
-                default -> Optional.empty();
-            };
+            var fieldBuilderTypeRef = generateBuilderTypeRef(field.type().value());
 
             if (fieldBuilderTypeRef.isPresent()) {
                 body.add(new JavaMethodDeclaration(
@@ -357,6 +353,14 @@ public class JavaTypesGenerator implements Generator {
         };
     }
 
+    private Optional<JavaTypeRef> generateBuilderTypeRef(Type type) {
+        return switch (type) {
+            case StructType structType -> Optional.of(generateBuilderTypeRef(structType));
+            case SumType sumType -> Optional.of(generateBuilderTypeRef(sumType));
+            default -> Optional.empty();
+        };
+    }
+
     private JavaTypeRef generateBuilderTypeRef(StructType type) {
         return generateBuilderTypeRef(type.namespaceName(), type.name());
     }