diff --git a/src/java-jpa-entity.js b/src/java-jpa-entity.js
index 392f17f..c9339c2 100644
--- a/src/java-jpa-entity.js
+++ b/src/java-jpa-entity.js
@@ -17,7 +17,7 @@ module.exports = {
                 
                 if(field.type.fullQualifiedName && field.type.fullQualifiedName.includes('models') && !field.isEnum) {
                     field.type.fullQualifiedName = field.type.fullQualifiedName.replace('models', suffix)
-                    field.type.imports = []
+                    field.type.imports = field.type.imports.filter(i => !i.includes('models'))
                 }
 
                 field.default = javaType.mapDefaultValue(field.type, field.default)