Fix test based on previous update