Repair broken test