Update integration and unit test

- Update integration and unit test for label_tool.