Breeding *Bug*

I created a system for my custom animal to be able to breed, I created 2 entities one in the adult variation and a baby one, using a procedure by despawning the child mob (adult) and spawning a mob in the baby variation, my problem is not with breeding but with with the AI ​​of the baby mob that doesn't follow its parents, and I can't find any solution online to make this mob in the baby variation follow its parents, because when the mob(recognized as a child) is despawned, its replacement doesn't recognize the 2 mobs that bred as parents, does anyone know a method to solve this problem?