diff --git a/posthog/hogql/test/test_bytecode.py b/posthog/hogql/test/test_bytecode.py
index cf0b8113b574d8..f7d810700e74a7 100644
--- a/posthog/hogql/test/test_bytecode.py
+++ b/posthog/hogql/test/test_bytecode.py
@@ -130,7 +130,7 @@ def test_bytecode_create(self):
     def test_bytecode_create_error(self):
         with self.assertRaises(NotImplementedException) as e:
             to_bytecode("(select 1)")
-        self.assertEqual(str(e.exception), "Visitor has no method visit_select_query")
+        self.assertEqual(str(e.exception), "BytecodeBuilder has no method visit_select_query")
 
         with self.assertRaises(NotImplementedException) as e:
             to_bytecode("1 in cohort 2")
diff --git a/posthog/hogql/test/test_visitor.py b/posthog/hogql/test/test_visitor.py
index 8aa6689328fbf3..a01193f788d5ff 100644
--- a/posthog/hogql/test/test_visitor.py
+++ b/posthog/hogql/test/test_visitor.py
@@ -125,7 +125,7 @@ def visit_arithmetic_operation(self, node: ast.ArithmeticOperation):
 
         with self.assertRaises(HogQLException) as e:
             UnknownNotDefinedVisitor().visit(parse_expr("1 + 3 / 'asd2'"))
-        self.assertEqual(str(e.exception), "Visitor has no method visit_constant")
+        self.assertEqual(str(e.exception), "UnknownNotDefinedVisitor has no method visit_constant")
 
     def test_hogql_exception_start_end(self):
         class EternalVisitor(TraversingVisitor):