From 630facb9fd487eb013241461daf20d55cae16c32 Mon Sep 17 00:00:00 2001
From: "Maxim.Medvedev" <maxim.medvedev@jetbrains.com>
Date: Mon, 4 Jun 2012 17:30:18 +0400
Subject: [PATCH] EA-36416 - PIEAE: LeafPsiElement.invalid

---
 .../groovy/lang/completion/weighers/GrWithWeigher.java        | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/completion/weighers/GrWithWeigher.java b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/completion/weighers/GrWithWeigher.java
index 7160d6c676e0..4625e123a102 100644
--- a/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/completion/weighers/GrWithWeigher.java
+++ b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/completion/weighers/GrWithWeigher.java
@@ -20,7 +20,7 @@ import com.intellij.codeInsight.completion.CompletionWeigher;
 import com.intellij.codeInsight.lookup.LookupElement;
 import com.intellij.psi.PsiElement;
 import org.jetbrains.annotations.NotNull;
-import org.jetbrains.plugins.groovy.lang.psi.GroovyFileBase;
+import org.jetbrains.plugins.groovy.GroovyFileType;
 import org.jetbrains.plugins.groovy.lang.psi.GroovyPsiElement;
 import org.jetbrains.plugins.groovy.lang.psi.api.GroovyResolveResult;
 import org.jetbrains.plugins.groovy.lang.psi.api.statements.expressions.GrReferenceExpression;
@@ -36,7 +36,7 @@ public class GrWithWeigher extends CompletionWeigher {
   @Override
   public Comparable weigh(@NotNull LookupElement element, @NotNull CompletionLocation location) {
     final PsiElement position = location.getCompletionParameters().getPosition();
-    if (!(position.getContainingFile() instanceof GroovyFileBase)) return null;
+    if (!(position.getLanguage() != GroovyFileType.GROOVY_LANGUAGE)) return null;
 
     if (!(position.getParent() instanceof GrReferenceExpression)) return null;