diff --git a/tests/pipeline/test_term.py b/tests/pipeline/test_term.py index 3f386907..7d630a01 100644 --- a/tests/pipeline/test_term.py +++ b/tests/pipeline/test_term.py @@ -369,7 +369,7 @@ class ObjectIdentityTestCase(TestCase): int_column = Column(dtype=int64_dtype, missing_value=3) self.assertTrue( - str(e.exception.message).startswith( + str(e.exception.args[0]).startswith( "Failed to create Column with name 'bad_column'" ) )