Fixed issue with category labels on update