diff --git a/python/src/com/jetbrains/python/psi/impl/PyReferenceExpressionImpl.java b/python/src/com/jetbrains/python/psi/impl/PyReferenceExpressionImpl.java
index 00b2126aed7c..b99d83a82ab8 100644
--- a/python/src/com/jetbrains/python/psi/impl/PyReferenceExpressionImpl.java
+++ b/python/src/com/jetbrains/python/psi/impl/PyReferenceExpressionImpl.java
@@ -60,7 +60,10 @@ public class PyReferenceExpressionImpl extends PyElementImpl implements PyRefere
   @NotNull
   @Override
   public PsiPolyVariantReference getReference() {
-    return getReference(PyResolveContext.defaultContext());
+    //noinspection InstanceofIncompatibleInterface
+    assert !(this instanceof StubBasedPsiElement);
+    final TypeEvalContext context = TypeEvalContext.codeAnalysis(getProject(), getContainingFile());
+    return getReference(PyResolveContext.defaultContext().withTypeEvalContext(context));
   }
 
   @NotNull