diff --git a/src/java-jpa-entity.js b/src/java-jpa-entity.js
index 4d102bb..72ddc12 100644
--- a/src/java-jpa-entity.js
+++ b/src/java-jpa-entity.js
@@ -14,7 +14,7 @@ module.exports = {
 
             entity.name = `${utils.pascalcase(model.name)}Record`
             entity.package = `${namespace}.${suffix}`;
-            entity.dir = model.package.replace(/\./g, '/')
+            entity.dir = entity.package.replace(/\./g, '/')
             entity.fields = entity.fields.map(field => {
 
                 let field_attr = field.attributes.find(attr => attr.name == 'jpa_sql')