From 0121686a7f9c44c543020c5a94fbafc3fa4f4dfc Mon Sep 17 00:00:00 2001
From: Kirill Rakhman <kirill.rakhman@jetbrains.com>
Date: Tue, 29 Aug 2023 17:03:42 +0200
Subject: [PATCH] [RAW FIR] Use KtFakeSourceElementKind.WrappedDelegate in LT

---
 .../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 61aa5f344fa..40486c3b5cc 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
@@ -1258,7 +1258,7 @@ class LightTreeRawFirDeclarationBuilder(
                 this.isLocal = true
                 val delegateBuilder = delegateExpression?.let {
                     FirWrappedDelegateExpressionBuilder().apply {
-                        source = delegateSource
+                        source = delegateSource?.fakeElement(KtFakeSourceElementKind.WrappedDelegate)
                         expression = expressionConverter.getAsFirExpression(it, "Incorrect delegate expression")
                     }
                 }
@@ -1283,7 +1283,7 @@ class LightTreeRawFirDeclarationBuilder(
 
                     val delegateBuilder = delegateExpression?.let {
                         FirWrappedDelegateExpressionBuilder().apply {
-                            source = delegateSource
+                            source = delegateSource?.fakeElement(KtFakeSourceElementKind.WrappedDelegate)
                             expression = expressionConverter.getAsFirExpression(it, "Should have delegate")
                         }
                     }