Fixed test code.