diff --git a/src/generators/java-jpa-entities.generator.js b/src/generators/java-jpa-entities.generator.js
index 1f91cd3..840a94d 100644
--- a/src/generators/java-jpa-entities.generator.js
+++ b/src/generators/java-jpa-entities.generator.js
@@ -11,18 +11,13 @@ module.exports = (service) => {
     const jpaModels = service.models.filter(model => model.attributes.find(attr => attr.name == 'jpa_sql'))
     const models = jpaEntites.mapJavaJpaEntity(jpaModels, service.namespace)
 
-    const modelFiles = models.map(model => {
+    return models.map(model => {
         return {
             name: `${model.name}.java`,
             dir: model.dir,
             contents: Mustache.render(templateEntities, { service, ...model })
         }
     })
-    return [
-        ...modelFiles,
-        ...enumFiles,
-        ...unionFiles
-    ]
 
 
 }
\ No newline at end of file