update tests for new labeling method