Fixing the category button