Fixed test case bug