Fixed test failture