Add the category to unit test for ease test