updated the unit test