diff --git a/java/kotlin-extractor2/src/main/kotlin/KotlinFileExtractor.kt b/java/kotlin-extractor2/src/main/kotlin/KotlinFileExtractor.kt
index 54574a20ddf..8cd58d90736 100644
--- a/java/kotlin-extractor2/src/main/kotlin/KotlinFileExtractor.kt
+++ b/java/kotlin-extractor2/src/main/kotlin/KotlinFileExtractor.kt
@@ -8819,7 +8819,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)