diff --git a/mapstruct-kotlin-processor/src/main/kotlin/com/github/pozo/BuilderProcessor.kt b/mapstruct-kotlin-processor/src/main/kotlin/com/github/pozo/BuilderProcessor.kt index 170d057..60934a6 100644 --- a/mapstruct-kotlin-processor/src/main/kotlin/com/github/pozo/BuilderProcessor.kt +++ b/mapstruct-kotlin-processor/src/main/kotlin/com/github/pozo/BuilderProcessor.kt @@ -73,6 +73,7 @@ class BuilderProcessor : AbstractProcessor() { val packageName = processingEnv.elementUtils.getPackageOf(typeElement).toString() val builderClassName = "${typeElement.simpleName}Builder" val classBuilder = TypeSpec.classBuilder(builderClassName) + .addOriginatingElement(typeElement) with(classBuilder) { this.addModifiers(Modifier.PUBLIC, Modifier.FINAL) @@ -110,4 +111,4 @@ class BuilderProcessor : AbstractProcessor() { override fun getSupportedOptions(): Set { return setOf("kapt.kotlin.generated") } -} \ No newline at end of file +}