Guardians floats/swim mid-air

Guardians can now fight water bubble current and sometimes randomly floats/swim mid-air.

Maybe because of the new 1.18 blocks that I've used in my build may have affected their AI/behvaior or,

Maybe after they're out of the water, their AI didn't end their swimming behavior.

See attached video below. Used entity x-ray to better see mobs through blocks.

I don't know how to reproduce the bug since I really don't know what causing this. Thanks devs for all your works ❤️