Fixed test broken by previous commit