diff --git a/python/psi-api/src/com/jetbrains/python/psi/PyClass.java b/python/psi-api/src/com/jetbrains/python/psi/PyClass.java
index 26553ee50951..c5054bee8cee 100644
--- a/python/psi-api/src/com/jetbrains/python/psi/PyClass.java
+++ b/python/psi-api/src/com/jetbrains/python/psi/PyClass.java
@@ -113,6 +113,9 @@ public interface PyClass extends PsiNameIdentifierOwner, PyStatement, NameDefine
   @NotNull
   List<PyClass> getAncestorClasses();
 
+  @NotNull
+  List<PyClass> getAncestorClasses(@NotNull TypeEvalContext context);
+
   /**
    * Get a list of all ancestor types.
    */
diff --git a/python/src/com/jetbrains/python/psi/impl/PyClassImpl.java b/python/src/com/jetbrains/python/psi/impl/PyClassImpl.java
index a9881353f251..1f0243284ebe 100644
--- a/python/src/com/jetbrains/python/psi/impl/PyClassImpl.java
+++ b/python/src/com/jetbrains/python/psi/impl/PyClassImpl.java
@@ -177,8 +177,14 @@ public class PyClassImpl extends PyPresentableElementImpl<PyClassStub> implement
   @NotNull
   @Override
   public List<PyClass> getAncestorClasses() {
+    return getAncestorClasses(TypeEvalContext.fastStubOnly(null));
+  }
+
+  @NotNull
+  @Override
+  public List<PyClass> getAncestorClasses(@NotNull TypeEvalContext context) {
     final List<PyClass> results = new ArrayList<PyClass>();
-    for (PyClassLikeType type : getAncestorTypes(TypeEvalContext.fastStubOnly(null))) {
+    for (PyClassLikeType type : getAncestorTypes(context)) {
       if (type instanceof PyClassType) {
         results.add(((PyClassType)type).getPyClass());
       }