Guardians Fly

We were building a guardian farm on Minecraft realms however we noticed a bug where guardians will begin to fly after leaving falling out of the water, I think this is a glitch between the AI wanting to go back into the water and still be stuck in swim mode.