refactor tests for labels