diff --git a/src/generators/java-jpa-entities.generator.js b/src/generators/java-jpa-entities.generator.js
index 865ce3f..1f91cd3 100644
--- a/src/generators/java-jpa-entities.generator.js
+++ b/src/generators/java-jpa-entities.generator.js
@@ -15,7 +15,7 @@ module.exports = (service) => {
         return {
             name: `${model.name}.java`,
             dir: model.dir,
-            contents: Mustache.render(templateModel, { service, ...model })
+            contents: Mustache.render(templateEntities, { service, ...model })
         }
     })
     return [