diff --git a/idea/src/org/jetbrains/jet/plugin/codeInsight/CodeInsightUtils.java b/idea/src/org/jetbrains/jet/plugin/codeInsight/CodeInsightUtils.java index 408d02fc12c..157820cf3b0 100644 --- a/idea/src/org/jetbrains/jet/plugin/codeInsight/CodeInsightUtils.java +++ b/idea/src/org/jetbrains/jet/plugin/codeInsight/CodeInsightUtils.java @@ -102,8 +102,16 @@ public class CodeInsightUtils { if (element1 == null || element2 == null) return null; - return PsiTreeUtil.findElementOfClassAtRange(file, element1.getTextRange().getStartOffset(), - element2.getTextRange().getEndOffset(), aClass); + startOffset = element1.getTextRange().getStartOffset(); + endOffset = element2.getTextRange().getEndOffset(); + + JetExpression jetExpression = PsiTreeUtil.findElementOfClassAtRange(file, startOffset, endOffset, aClass); + if (jetExpression == null || + jetExpression.getTextRange().getStartOffset() != startOffset || + jetExpression.getTextRange().getEndOffset() != endOffset) { + return null; + } + return jetExpression; } @Nullable