diff --git a/platform/lang-api/src/com/intellij/codeInspection/ProblemsHolder.java b/platform/lang-api/src/com/intellij/codeInspection/ProblemsHolder.java
index f8e4cee0e030..0c27dc3f7521 100644
--- a/platform/lang-api/src/com/intellij/codeInspection/ProblemsHolder.java
+++ b/platform/lang-api/src/com/intellij/codeInspection/ProblemsHolder.java
@@ -27,6 +27,7 @@ import com.intellij.psi.PsiElement;
 import com.intellij.psi.PsiFile;
 import com.intellij.psi.PsiReference;
 import com.intellij.psi.util.PsiTreeUtil;
+import com.intellij.util.ArrayUtil;
 import org.jetbrains.annotations.Nls;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
@@ -90,7 +91,7 @@ public class ProblemsHolder {
 
   private boolean isInPsiFile(@NotNull PsiElement element) {
     PsiFile file = element.getContainingFile();
-    return myFile.getViewProvider().getAllFiles().contains(file);
+    return ArrayUtil.indexOf(myFile.getPsiRoots(), file) != -1;
   }
 
   private void redirectProblem(@NotNull final ProblemDescriptor problem, @NotNull final PsiElement target) {