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)
         }
     }