diff --git a/src/java-type.js b/src/java-type.js
index 7545802..eb8007d 100644
--- a/src/java-type.js
+++ b/src/java-type.js
@@ -54,7 +54,7 @@ function getCustomType(type, namespace) {
     let match = LAST_ELEMENT_REGEX.exec(type)
     let name = ''
     let importstatement = null
-    let isEnum = name.includes('enum')
+    let isEnum = type.includes('enums')
     if(match) {
         name = utils.pascalcase(match[2])
         importstatement = `${match[1].replace('enums', 'models')}.${name}`