diff --git a/src/java-type.js b/src/java-type.js
index 3ad6a7f..3fdaa8a 100644
--- a/src/java-type.js
+++ b/src/java-type.js
@@ -52,10 +52,11 @@ function getCustomType(type, namespace) {
     let importstatement = null
     if(match) {
         name = utils.pascalcase(match[2])
-        importstatement = `${match[1]}.${name}`
+
+        importstatement = `${match[1].replace('enums', 'models')}.${name}`
     }else {
         name = utils.pascalcase(type)
-        // importstatement = `${namespace}.models.${name}`
+        importstatement = `${namespace}.models.${name}`
     }
     
     return {