diff --git a/java/kotlin-extractor2/src/main/kotlin/KotlinFileExtractor.kt b/java/kotlin-extractor2/src/main/kotlin/KotlinFileExtractor.kt
index 9065e2548bd..9d734d774fc 100644
--- a/java/kotlin-extractor2/src/main/kotlin/KotlinFileExtractor.kt
+++ b/java/kotlin-extractor2/src/main/kotlin/KotlinFileExtractor.kt
@@ -8820,7 +8820,7 @@ OLD: KE1
             tw.writeHasLocation(id, locId)
 
             // Extract constructor
-            val unitType = useType(pluginContext.irBuiltIns.unitType)
+            val unitType = useType(pluginContext.irBuiltIns.unitType, TypeContext.RETURN)
             tw.writeConstrs(ids.constructor, "", "", unitType.javaResult.id, id, ids.constructor)
             tw.writeConstrsKotlinType(ids.constructor, unitType.kotlinResult.id)
             tw.writeHasLocation(ids.constructor, locId)