diff --git a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt
index 183d27fe337..20d75c1fd5c 100644
--- a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt
+++ b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt
@@ -380,6 +380,7 @@ private class ContextCollectorVisitor(
                 try {
                     smartCasts = smartCasts.put(symbol, smartCastExpression.typesFromSmartCast.toSet())
                     super.visitSmartCastExpression(smartCastExpression)
+                    return
                 } finally {
                     smartCasts = previousSmartCasts
                 }