modified the unit test as required