Fixed test after updating