Fixed test case failure