From 128075e7809bd242dcba40a36bc465af7ad40e6e Mon Sep 17 00:00:00 2001
From: Oleg Ivanov <oleg.ivanov@jetbrains.com>
Date: Thu, 16 Jul 2020 15:42:33 +0300
Subject: [PATCH] [FIR] Add fir source saving in resolved contract description

---
 .../transformers/contracts/FirContractResolveTransformer.kt      | 1 +
 1 file changed, 1 insertion(+)

diff --git a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/contracts/FirContractResolveTransformer.kt b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/contracts/FirContractResolveTransformer.kt
index f763ba98c43..4bc3fbb61e9 100644
--- a/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/contracts/FirContractResolveTransformer.kt
+++ b/compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/contracts/FirContractResolveTransformer.kt
@@ -140,6 +140,7 @@ class FirContractResolveTransformer(
                         effects += effect
                     }
                 }
+                this.source = owner.contractDescription.source
             }
             owner.replaceContractDescription(resolvedContractDescription)
             dataFlowAnalyzer.exitContractDescription()