improve category_id in schema