Fixed test cases to the changes made