fix tests for new labels