changed the unit test