From 0fdd33b415f161ea095f03ea3dc92a1ae2d14e6e Mon Sep 17 00:00:00 2001
From: Max Medvedev <maxim.medvedev@jetbrains.com>
Date: Fri, 23 Nov 2012 15:23:10 +0400
Subject: [PATCH] correct package definition search

---
 .../jetbrains/plugins/groovy/lang/psi/impl/GroovyFileImpl.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/impl/GroovyFileImpl.java b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/impl/GroovyFileImpl.java
index 5f0167ee457a..2b8aabb0c8d6 100644
--- a/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/impl/GroovyFileImpl.java
+++ b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/impl/GroovyFileImpl.java
@@ -52,6 +52,7 @@ import org.jetbrains.plugins.groovy.lang.psi.api.types.GrCodeReferenceElement;
 import org.jetbrains.plugins.groovy.lang.psi.impl.synthetic.GrLightParameter;
 import org.jetbrains.plugins.groovy.lang.psi.impl.synthetic.GroovyScriptClass;
 import org.jetbrains.plugins.groovy.lang.psi.stubs.GrFileStub;
+import org.jetbrains.plugins.groovy.lang.psi.stubs.GrPackageDefinitionStub;
 import org.jetbrains.plugins.groovy.lang.resolve.MethodTypeInferencer;
 import org.jetbrains.plugins.groovy.lang.resolve.ResolveUtil;
 import org.jetbrains.plugins.groovy.lang.resolve.processors.ClassHint;
@@ -99,7 +100,7 @@ public class GroovyFileImpl extends GroovyFileBaseImpl implements GroovyFile {
     final StubElement<?> stub = getStub();
     if (stub != null) {
       for (StubElement element : stub.getChildrenStubs()) {
-        if (element instanceof GrPackageDefinition) return (GrPackageDefinition)element;
+        if (element instanceof GrPackageDefinitionStub) return (GrPackageDefinition)element.getPsi();
       }
       return null;
     }