Fixed test following previous patch