Modified test of LabelAssignerTest