Fixed test to pass