diff --git a/pybind11_protobuf/tests/pass_by_test.py b/pybind11_protobuf/tests/pass_by_test.py
index 49cc1a6..0ced2f9 100644
--- a/pybind11_protobuf/tests/pass_by_test.py
+++ b/pybind11_protobuf/tests/pass_by_test.py
@@ -113,8 +113,7 @@ def test_py_proto_check(self, check_method):
   @parameterized.named_parameters(get_pass_by_params())
   def test_pool_proto_check(self, check_method):
     pool = descriptor_pool.Default()
-    factory = message_factory.MessageFactory(pool)
-    prototype = factory.GetPrototype(
+    prototype = message_factory.GetMessageClass(
         pool.FindMessageTypeByName('pybind11.test.IntMessage')
     )
     message = prototype(value=9)