Fixed test code I lefted