From 7087f10660a9d58b028ce47c65d2b64aafef1bc5 Mon Sep 17 00:00:00 2001
From: Natalia Ukhorskaya <natalia.ukhorskaya@jetbrains.com>
Date: Wed, 3 Aug 2016 17:01:11 +0300
Subject: [PATCH] Fixes after review: get last statement in codeFragmetn using
 psi instead of findElementAt

---
 .../kotlin/idea/debugger/evaluate/KotlinCodeFragmentFactory.kt  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/idea/src/org/jetbrains/kotlin/idea/debugger/evaluate/KotlinCodeFragmentFactory.kt b/idea/src/org/jetbrains/kotlin/idea/debugger/evaluate/KotlinCodeFragmentFactory.kt
index c94dc7f8af5..91896f84b6d 100644
--- a/idea/src/org/jetbrains/kotlin/idea/debugger/evaluate/KotlinCodeFragmentFactory.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/debugger/evaluate/KotlinCodeFragmentFactory.kt
@@ -412,6 +412,6 @@ class KotlinCodeFragmentFactory: CodeFragmentFactory() {
             }
         })
 
-        return getContextElement(codeFragment.findElementAt(codeFragment.text.length - 5)) as? KtElement
+        return codeFragment.getContentElement().statements.lastOrNull()
     }
 }