Fix tests for AssertionError output