[FIX] Assign correct category