Fixed test code lying arund