diff --git a/python/src/com/jetbrains/python/pyi/PyiTypeProvider.java b/python/src/com/jetbrains/python/pyi/PyiTypeProvider.java
index e4edb9daeada..221817a5f578 100644
--- a/python/src/com/jetbrains/python/pyi/PyiTypeProvider.java
+++ b/python/src/com/jetbrains/python/pyi/PyiTypeProvider.java
@@ -172,7 +172,7 @@ public class PyiTypeProvider extends PyTypeProviderBase {
     if (owner instanceof PyClass) {
       final PyClass cls = (PyClass)owner;
       if (name != null) {
-        cls.visitMethods(overloadsProcessor, false);
+        cls.visitMethods(overloadsProcessor, false, context);
       }
     }
     else if (owner instanceof PyFile) {