From d675c12a32e5f7208688a38ff14405d02b8cd9ea Mon Sep 17 00:00:00 2001
From: Natalia Ukhorskaya <natalia.ukhorskaya@jetbrains.com>
Date: Wed, 19 Aug 2015 13:57:56 +0300
Subject: [PATCH] Minor: change method return type

---
 .../idea/debugger/evaluate/KotlinCodeFragmentFactory.kt     | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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 ba4dc01c74d..77f6dcec40f 100644
--- a/idea/src/org/jetbrains/kotlin/idea/debugger/evaluate/KotlinCodeFragmentFactory.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/debugger/evaluate/KotlinCodeFragmentFactory.kt
@@ -127,7 +127,7 @@ class KotlinCodeFragmentFactory: CodeFragmentFactory() {
         public val LABEL_VARIABLE_VALUE_KEY: Key<Value> = Key.create<Value>("_label_variable_value_key_")
         public val DEBUG_LABEL_SUFFIX: String = "_DebugLabel"
 
-        fun getContextElement(elementAt: PsiElement?): PsiElement? {
+        fun getContextElement(elementAt: PsiElement?): JetElement? {
             if (elementAt == null) return null
 
             if (elementAt is PsiCodeBlock) {
@@ -170,7 +170,7 @@ class KotlinCodeFragmentFactory: CodeFragmentFactory() {
         }
     }
 
-    private fun wrapContextIfNeeded(project: Project, originalContext: PsiElement?): PsiElement? {
+    private fun wrapContextIfNeeded(project: Project, originalContext: JetElement?): JetElement? {
         val session = XDebuggerManager.getInstance(project).getCurrentSession() as? XDebugSessionImpl
                                             ?: return originalContext
 
@@ -189,7 +189,7 @@ class KotlinCodeFragmentFactory: CodeFragmentFactory() {
             labels: Map<String, ObjectReference>,
             originalContext: PsiElement?,
             project: Project
-    ): PsiElement? {
+    ): JetElement? {
         val codeFragment = JetPsiFactory(project).createBlockCodeFragment(newFragmentText, originalContext)
 
         codeFragment.accept(object : JetTreeVisitorVoid() {