From dc19e39dd28286c5919d5234e9fa54de4cef63af Mon Sep 17 00:00:00 2001
From: Andrey Zinovyev <andrey.zinovyev@jetbrains.com>
Date: Wed, 14 Apr 2021 20:12:05 +0300
Subject: [PATCH] [FIR][LightTree] Properly locate return keyword

---
 .../fir/analysis/diagnostics/LightTreePositioningStrategies.kt  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/diagnostics/LightTreePositioningStrategies.kt b/compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/diagnostics/LightTreePositioningStrategies.kt
index 6294d997a6c..b19a4e37e80 100644
--- a/compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/diagnostics/LightTreePositioningStrategies.kt
+++ b/compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/diagnostics/LightTreePositioningStrategies.kt
@@ -545,7 +545,7 @@ object LightTreePositioningStrategies {
             if (labeledExpression != null) {
                 return markRange(node, labeledExpression, startOffset, endOffset, tree, node)
             }
-            return markElement(tree.returnKeyword(node) ?: node, startOffset, endOffset, tree)
+            return markElement(tree.returnKeyword(node) ?: node, startOffset, endOffset, tree, node)
         }
     }