Preserve CategoricalComponent information in custom viewers