Fixed issue with category