Fixed test! :)