Fixed test after the previous commit