From 6ce6301613565d9122af58f70c2749b55b3a16e2 Mon Sep 17 00:00:00 2001
From: "Vitaliy.Bibaev" <vitaliy.bibaev@jetbrains.com>
Date: Thu, 26 Oct 2017 15:19:12 +0300
Subject: [PATCH] Fix bug when chain was not found from nested lambda or
 anonymous

---
 .../streams/kotlin/psi/impl/KotlinChainBuilderBase.kt         | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/idea/idea-jvm/src/com/intellij/debugger/streams/kotlin/psi/impl/KotlinChainBuilderBase.kt b/idea/idea-jvm/src/com/intellij/debugger/streams/kotlin/psi/impl/KotlinChainBuilderBase.kt
index 9cbbd90a475..4497601707d 100644
--- a/idea/idea-jvm/src/com/intellij/debugger/streams/kotlin/psi/impl/KotlinChainBuilderBase.kt
+++ b/idea/idea-jvm/src/com/intellij/debugger/streams/kotlin/psi/impl/KotlinChainBuilderBase.kt
@@ -34,7 +34,7 @@ abstract class KotlinChainBuilderBase(private val transformer: ChainTransformer<
     var element = getLatestElementInScope(start)
     while (element != null) {
       element.accept(visitor)
-      element = getLatestElementInScope(toUpperLevel(element))
+      element = toUpperLevel(element)
     }
 
     return visitor.chains().map { transformer.transform(it, startElement) }
@@ -47,7 +47,7 @@ abstract class KotlinChainBuilderBase(private val transformer: ChainTransformer<
       current = current.parent
     }
 
-    return current
+    return getLatestElementInScope(current)
   }
 
   protected abstract fun createChainsBuilder(): ChainBuilder