From a849cc7da495580443d2479f086fecdd638bf259 Mon Sep 17 00:00:00 2001
From: Simon Ogorodnik <simon.ogorodnik@jetbrains.com>
Date: Mon, 20 May 2019 19:55:50 +0300
Subject: [PATCH] Store type for already resolved FIR references properly

---
 .../fir/resolve/transformers/FirBodyResolveTransformer.kt      | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirBodyResolveTransformer.kt b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirBodyResolveTransformer.kt
index 19d77d365a4..aa46a19451d 100644
--- a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirBodyResolveTransformer.kt
+++ b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirBodyResolveTransformer.kt
@@ -406,8 +406,7 @@ open class FirBodyResolveTransformer(val session: FirSession, val implicitTypeOn
             }
             is FirResolvedCallableReference -> {
                 if (qualifiedAccessExpression.typeRef !is FirResolvedTypeRef) {
-                    qualifiedAccessExpression.resultType =
-                        jump.tryCalculateReturnType(callee.coneSymbol.firUnsafe<FirCallableDeclaration>())
+                    storeTypeFromCallee(qualifiedAccessExpression)
                 }
             }
         }