updating the unit test cases