Add category back