diff --git a/pybind11_protobuf/tests/message_test.py b/pybind11_protobuf/tests/message_test.py
index 033f879..a92caea 100644
--- a/pybind11_protobuf/tests/message_test.py
+++ b/pybind11_protobuf/tests/message_test.py
@@ -436,11 +436,15 @@ def test_text_format_to_string(self):
 
   def test_text_format_parse(self):
     message = text_format.Parse(m.TEXT_FORMAT_MESSAGE, m.make_test_message())
-    self.assertMultiLineEqual(str(message), m.TEXT_FORMAT_MESSAGE)
+    self.assertMultiLineEqual(
+        text_format.MessageToString(message), m.TEXT_FORMAT_MESSAGE
+    )
 
   def test_text_format_merge(self):
     message = text_format.Merge(m.TEXT_FORMAT_MESSAGE, m.make_test_message())
-    self.assertMultiLineEqual(str(message), m.TEXT_FORMAT_MESSAGE)
+    self.assertMultiLineEqual(
+        text_format.MessageToString(message), m.TEXT_FORMAT_MESSAGE
+    )
 
   def test_proto_2_equal(self):
     self.assertProtoEqual(m.TEXT_FORMAT_MESSAGE,