From e42e923d4e065a87d2081888d2230290730e54f3 Mon Sep 17 00:00:00 2001
From: "Pavel V. Talanov" <talanov.pavel@gmail.com>
Date: Thu, 17 Apr 2014 14:44:20 +0400
Subject: [PATCH] JetPackageDirective#getPackageNameExpression() by stub

---
 .../src/org/jetbrains/jet/lang/psi/JetPackageDirective.java    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPackageDirective.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPackageDirective.java
index f63be6d2116..010f32976f2 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPackageDirective.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPackageDirective.java
@@ -18,6 +18,7 @@ package org.jetbrains.jet.lang.psi;
 
 import com.intellij.lang.ASTNode;
 import com.intellij.psi.PsiElement;
+import com.intellij.util.ArrayFactory;
 import com.intellij.util.containers.ContainerUtil;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
@@ -45,7 +46,7 @@ public class JetPackageDirective extends JetExpressionImplStub<PsiJetPlaceHolder
     // This should be either JetSimpleNameExpression, or JetDotQualifiedExpression
     @Nullable
     public JetExpression getPackageNameExpression() {
-        return findChildByClass(JetExpression.class);
+        return JetStubbedPsiUtil.getStubOrPsiChild(this, JetImportDirective.IMPORT_DIRECTIVE_EXPRESSIONS, JetExpression.ARRAY_FACTORY);
     }
 
     @NotNull