PIM-2672 : use the related entity in the category extension