Another bug fix with updating categories