Fixed test, now working!