How would i make a floatie on a swimming NPC work correctly?

[Saffwat (Safwart)]

i want my npc to float like the second image instead of having 90% of its body underwater when swimming


(this guy is anchored)