diff --git a/src/main/resources/JavaLombok/modelOneOf.handlebars b/src/main/resources/JavaLombok/modelOneOf.handlebars
index ea7f83f..2d9f114 100644
--- a/src/main/resources/JavaLombok/modelOneOf.handlebars
+++ b/src/main/resources/JavaLombok/modelOneOf.handlebars
@@ -2,8 +2,7 @@ public sealed interface {{classname}}
  	{{~#interfaces}}{{#if @first}} extends {{else}}, {{/if}}{{{.}}}{{/interfaces}}
  	{{~#children}}{{#if @first}} permits {{else}}, {{/if}}{{{classname}}}{{/children}} {
 
-{{#if discriminator}}
-{{#discriminator}}
+{{#with discriminator}}
     {{{propertyType}}} {{propertyGetter}}();
 
     static Class<? extends {{{classname}}}> getType(String {{propertyName}}) {
@@ -20,6 +19,5 @@ public sealed interface {{classname}}
         return new {{{modelName}}}().set{{capitalizeFirst propertyName}}("{{mappingName}}");
     }
     {{/mappedModels}}
-{{/discriminator}}
-{{/if}}
+{{/with}}
 }
\ No newline at end of file