From 18ad56f7a8f3d34fefe1cfad9cd5f8c381d2d30b Mon Sep 17 00:00:00 2001
From: Dmitriy Novozhilov <dmitriy.novozhilov@jetbrains.com>
Date: Fri, 28 Aug 2020 14:45:47 +0300
Subject: [PATCH] [FIR] Fix creating fake source elements in declarations
 resolve

---
 .../body/resolve/FirDeclarationsResolveTransformer.kt     | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/body/resolve/FirDeclarationsResolveTransformer.kt b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/body/resolve/FirDeclarationsResolveTransformer.kt
index 02b2a37329d..e14ada4a702 100644
--- a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/body/resolve/FirDeclarationsResolveTransformer.kt
+++ b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/body/resolve/FirDeclarationsResolveTransformer.kt
@@ -895,8 +895,8 @@ open class FirDeclarationsResolveTransformer(transformer: FirBodyResolveTransfor
                             buildResolvedTypeRef {
                                 type = resultType.coneType
                                 annotations.addAll(resultType.annotations)
-                                resultType.source.psi?.let {
-                                    source = FirFakeSourceElement(it, FirFakeSourceElementKind.PropertyFromParameter)
+                                resultType.source?.fakeElement(FirFakeSourceElementKind.PropertyFromParameter)?.let {
+                                    source = it
                                 }
                             }
                         }
@@ -916,8 +916,8 @@ open class FirDeclarationsResolveTransformer(transformer: FirBodyResolveTransfor
                                 buildResolvedTypeRef {
                                     type = resultType.coneType
                                     annotations.addAll(resultType.annotations)
-                                    resultType.source.psi?.let {
-                                        source = FirFakeSourceElement(it, FirFakeSourceElementKind.PropertyFromParameter)
+                                    resultType.source?.fakeElement(FirFakeSourceElementKind.PropertyFromParameter)?.let {
+                                        source = it
                                     }
                                 }
                             }