Fixed test code