diff --git a/tests/stub/iteration/test_result_single.py b/tests/stub/iteration/test_result_single.py
index 58e10df2b..c1d71c547 100644
--- a/tests/stub/iteration/test_result_single.py
+++ b/tests/stub/iteration/test_result_single.py
@@ -15,8 +15,10 @@ def _assert_not_exactly_one_record_error(self, error):
         self.assertIsInstance(error, types.DriverError)
         driver = get_driver_name()
         if driver in ["python"]:
-            self.assertEqual("<class 'ToBeDecided'>",
-                             error.errorType)
+            self.assertEqual(
+                "<class 'neo4j.exceptions.ResultNotSingleError'>",
+                error.errorType
+            )
         elif driver in ["ruby"]:
             self.assertEqual(
                 "Neo4j::Driver::Exceptions::NoSuchRecordException",