Baby Hoglin/Zoglin heads come off during attack animation

When a Baby Hoglin or Zoglin attacks an entity, their head flies off during their animation. It's easier to see if you place an armor stand and get a baby zoglin to target it.