diff --git a/custom-generator/src/main/java/io/serverlessworkflow/generator/AllAnyOneOfSchemaRule.java b/custom-generator/src/main/java/io/serverlessworkflow/generator/AllAnyOneOfSchemaRule.java
index df3ead34..d14ba357 100644
--- a/custom-generator/src/main/java/io/serverlessworkflow/generator/AllAnyOneOfSchemaRule.java
+++ b/custom-generator/src/main/java/io/serverlessworkflow/generator/AllAnyOneOfSchemaRule.java
@@ -188,7 +188,7 @@ public JType apply(
           throw new IllegalStateException(ex);
         }
       }
-      schema.setJavaTypeIfEmpty(javaType);
+      schema.setJavaType(javaType);
     }
     return javaType;
   }