fix category_generator