Updated unit test for consistency