Zombified_Piglin, Drowned and Zoglin Team problem

 When a piglin becomes a zombified_piglin, a hoglin a zoglin and a zombie a drowned and they have a predefined team, after their transformations they lose their teams and stay without team.