diff --git a/plugins/kapt4/src/org/jetbrains/kotlin/kapt4/StubGenerator.kt b/plugins/kapt4/src/org/jetbrains/kotlin/kapt4/StubGenerator.kt
index 82c99f681aa..f017d3ba53d 100644
--- a/plugins/kapt4/src/org/jetbrains/kotlin/kapt4/StubGenerator.kt
+++ b/plugins/kapt4/src/org/jetbrains/kotlin/kapt4/StubGenerator.kt
@@ -424,7 +424,7 @@ private class StubGenerator(
             private fun elementMapping(lightClass: PsiClass): Multimap<KtElement, PsiElement> =
                 HashMultimap.create<KtElement, PsiElement>().apply {
                     (lightClass.methods.asSequence() + lightClass.fields.asSequence() + lightClass.constructors.asSequence()).forEach {
-                        (it as KtLightElement<*, *>).kotlinOrigin?.let { origin -> put(origin, it) }
+                        put((it as KtLightElement<*, *>).kotlinOrigin, it)
                     }
                 }