diff --git a/platform/lang-impl/src/com/intellij/psi/util/proximity/SameDirectoryWeigher.java b/platform/lang-impl/src/com/intellij/psi/util/proximity/SameDirectoryWeigher.java
index 7029551548a1..ecc3be1b2fa4 100644
--- a/platform/lang-impl/src/com/intellij/psi/util/proximity/SameDirectoryWeigher.java
+++ b/platform/lang-impl/src/com/intellij/psi/util/proximity/SameDirectoryWeigher.java
@@ -39,7 +39,7 @@ public class SameDirectoryWeigher extends ProximityWeigher {
   private static PsiDirectory getParentDirectory(PsiElement element) {
     PsiFile file = InjectedLanguageUtil.getTopLevelFile(element);
     if (file != null) {
-      element = file;
+      element = file.getOriginalFile();
     }
     while (element != null && !(element instanceof PsiDirectory)) {
       element = element.getParent();
diff --git a/platform/lang-impl/src/com/intellij/psi/util/proximity/SameLogicalRootWeigher.java b/platform/lang-impl/src/com/intellij/psi/util/proximity/SameLogicalRootWeigher.java
index 2e251dbebb16..78d1b26cf5af 100644
--- a/platform/lang-impl/src/com/intellij/psi/util/proximity/SameLogicalRootWeigher.java
+++ b/platform/lang-impl/src/com/intellij/psi/util/proximity/SameLogicalRootWeigher.java
@@ -57,7 +57,7 @@ public class SameLogicalRootWeigher extends ProximityWeigher {
     final PsiFile psiFile = element.getContainingFile();
     if (psiFile == null) return null;
 
-    final VirtualFile file = psiFile.getVirtualFile();
+    final VirtualFile file = psiFile.getOriginalFile().getVirtualFile();
     if (file == null) return null;
 
     return LogicalRootsManager.getLogicalRootsManager(element.getProject()).findLogicalRoot(file);