Fixed an issue with the category wrapper