diff --git a/idea/src/org/jetbrains/jet/plugin/search/KotlinReferencesSearcher.java b/idea/src/org/jetbrains/jet/plugin/search/KotlinReferencesSearcher.java index ffb166ab29b..5d2631bff48 100644 --- a/idea/src/org/jetbrains/jet/plugin/search/KotlinReferencesSearcher.java +++ b/idea/src/org/jetbrains/jet/plugin/search/KotlinReferencesSearcher.java @@ -16,7 +16,9 @@ package org.jetbrains.jet.plugin.search; +import com.intellij.openapi.application.ApplicationManager; import com.intellij.openapi.application.QueryExecutorBase; +import com.intellij.openapi.util.Computable; import com.intellij.psi.PsiElement; import com.intellij.psi.PsiMethod; import com.intellij.psi.PsiReference; @@ -46,7 +48,12 @@ public class KotlinReferencesSearcher extends QueryExecutorBase() { + @Override + public PsiMethod compute() { + return JetLightClass.wrapMethod(function); + } + }); if (method != null) { queryParameters.getOptimizer().searchWord(name, queryParameters.getScope(), true, JetLightClass.wrapMethod((JetFunction) element));