Fixed test following Mike's comment