diff --git a/test/test_plugin/test_stub.py b/test/test_plugin/test_stub.py
index d2b9084c..89377b92 100644
--- a/test/test_plugin/test_stub.py
+++ b/test/test_plugin/test_stub.py
@@ -73,4 +73,4 @@ def test_method(Script):
     def_, = Script(code).goto_definitions()
     context = def_._name._context
     assert isinstance(context, BoundMethod), context
-    assert isinstance(context.function, typeshed.ClassStubContext), context
+    assert isinstance(context._function, typeshed.FunctionStubContext), context