Removed AssertionError check and check for AssertionError in the test