Fixed test cases. No change to the code base