Correction to make mutation tests pass