diff --git a/python/src/com/jetbrains/python/psi/types/PyModuleType.java b/python/src/com/jetbrains/python/psi/types/PyModuleType.java
index 2cb8d13feb9f..74e75957e946 100644
--- a/python/src/com/jetbrains/python/psi/types/PyModuleType.java
+++ b/python/src/com/jetbrains/python/psi/types/PyModuleType.java
@@ -139,8 +139,11 @@ public class PyModuleType implements PyType { // Modules don't descend from obje
       }
     }
     if (!ResolveImportUtil.isAbsoluteImportEnabledFor(element)) {
-      final PyQualifiedName absoluteQName = ResolveImportUtil.findCanonicalImportPath(element, null);
-      final PsiFile file = element.getContainingFile();
+      PsiFile file = element.getContainingFile();
+      if (file != null) {
+        file = file.getOriginalFile();
+      }
+      final PyQualifiedName absoluteQName = ResolveImportUtil.findShortestImportableQName(file);
       if (file != null && absoluteQName != null) {
         final PyQualifiedName prefixQName = PyUtil.isPackage(file) ? absoluteQName : absoluteQName.removeLastComponent();
         if (prefixQName.getComponentCount() > 0) {