category assignment fix --KW