Fix issue with category import