modified the unit testcases