diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPsiUtil.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPsiUtil.java index 6f26c31a31c..aec5d03b885 100644 --- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPsiUtil.java +++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPsiUtil.java @@ -399,14 +399,8 @@ public class JetPsiUtil { @Nullable public static JetSimpleNameExpression getLastReference(@NotNull JetExpression importedReference) { - if (importedReference instanceof JetDotQualifiedExpression) { - JetExpression selectorExpression = ((JetDotQualifiedExpression) importedReference).getSelectorExpression(); - return (selectorExpression instanceof JetSimpleNameExpression) ? (JetSimpleNameExpression) selectorExpression : null; - } - if (importedReference instanceof JetSimpleNameExpression) { - return (JetSimpleNameExpression) importedReference; - } - return null; + JetElement selector = PsiUtilPackage.getQualifiedElementSelector(importedReference); + return selector instanceof JetSimpleNameExpression ? (JetSimpleNameExpression) selector : null; } public static boolean isSelectorInQualified(@NotNull JetSimpleNameExpression nameExpression) { diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/psiUtil/jetPsiUtil.kt b/compiler/frontend/src/org/jetbrains/jet/lang/psi/psiUtil/jetPsiUtil.kt index 3e5fd21f5d1..9b6a81a082e 100644 --- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/psiUtil/jetPsiUtil.kt +++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/psiUtil/jetPsiUtil.kt @@ -242,7 +242,7 @@ fun JetElement.getQualifiedElementSelector(): JetElement? { if (selector is JetCallExpression) selector.getCalleeExpression() else selector } is JetUserType -> getReferenceExpression() - else -> this + else -> null } }