diff --git a/src/java-type.js b/src/java-type.js index 87a25fd..9bbd485 100644 --- a/src/java-type.js +++ b/src/java-type.js @@ -56,7 +56,7 @@ function getCustomType(type, namespace, suffix) { let name = '' let importstatement = null if(match) { - name = utils.pascalcase(match[2]) + name = utils.pascalcase(match[2])+suffix importstatement = `${match[1].replace('enums', 'models')}.${name}${suffix}` }else { name = utils.pascalcase(type)