Iron Golem Visibility Inconsistent in Water

When an iron golem is pushed into water, it loses visibility as it sinks. Once underwater, its visibility phases in and out while player is above water. Not sure if other mobs are affected since they can float/swim.