diff --git a/src/generators/java-jpa-entities.generator.js b/src/generators/java-jpa-entities.generator.js
index 1465eb8..d6a2a8b 100644
--- a/src/generators/java-jpa-entities.generator.js
+++ b/src/generators/java-jpa-entities.generator.js
@@ -32,7 +32,7 @@ module.exports = (service) => {
         return {
             name: `${repo.name}.java`,
             dir: repo.dir,
-            contents: Mustache.render(getRepositoryTemplate(repo.type), { service, ...model })
+            contents: Mustache.render(getRepositoryTemplate(repo.type), { service, ...repo })
         }
     });