From c31d860f100f6a5e77e726d2983b63bb044ad9de Mon Sep 17 00:00:00 2001
From: Dmitrii Gridin <dmitry.gridin@jetbrains.com>
Date: Thu, 23 Nov 2023 11:46:54 +0100
Subject: [PATCH] [FIR] LightTreeRawFirBuilder: add copyWithNewSourceKind for
 generated copy

This is done for consistency

^KT-63042
---
 .../lightTree/converter/LightTreeRawFirDeclarationBuilder.kt  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/compiler/fir/raw-fir/light-tree2fir/src/org/jetbrains/kotlin/fir/lightTree/converter/LightTreeRawFirDeclarationBuilder.kt b/compiler/fir/raw-fir/light-tree2fir/src/org/jetbrains/kotlin/fir/lightTree/converter/LightTreeRawFirDeclarationBuilder.kt
index c35c13bf7ce..b8ca01857cd 100644
--- a/compiler/fir/raw-fir/light-tree2fir/src/org/jetbrains/kotlin/fir/lightTree/converter/LightTreeRawFirDeclarationBuilder.kt
+++ b/compiler/fir/raw-fir/light-tree2fir/src/org/jetbrains/kotlin/fir/lightTree/converter/LightTreeRawFirDeclarationBuilder.kt
@@ -589,8 +589,8 @@ class LightTreeRawFirDeclarationBuilder(
                             zippedParameters,
                             context.packageFqName,
                             context.className,
-                            createClassTypeRefWithSourceKind = { firPrimaryConstructor.returnTypeRef },
-                            createParameterTypeRefWithSourceKind = { property, _ -> property.returnTypeRef },
+                            createClassTypeRefWithSourceKind = { firPrimaryConstructor.returnTypeRef.copyWithNewSourceKind(it) },
+                            createParameterTypeRefWithSourceKind = { property, kind -> property.returnTypeRef.copyWithNewSourceKind(kind) },
                             addValueParameterAnnotations = { valueParam ->
                                 valueParam.forEachChildren {
                                     if (it.tokenType == MODIFIER_LIST) convertModifierList(it).annotations.filterTo(annotations) {