diff --git a/python/psi-api/src/com/jetbrains/python/psi/types/TypeEvalContext.java b/python/psi-api/src/com/jetbrains/python/psi/types/TypeEvalContext.java
index 2a5c2b475f5d..03bb1ac60102 100644
--- a/python/psi-api/src/com/jetbrains/python/psi/types/TypeEvalContext.java
+++ b/python/psi-api/src/com/jetbrains/python/psi/types/TypeEvalContext.java
@@ -159,7 +159,7 @@ public class TypeEvalContext {
   }
 
   public boolean maySwitchToAST(@NotNull StubBasedPsiElement element) {
-    return myAllowStubToAST || (element.getStub() == null && (myOrigin == null || myOrigin == element.getContainingFile()));
+    return myAllowStubToAST || myOrigin == element.getContainingFile();
   }
 
   @Nullable