Enderdragon often flies "out of bounds" and glitches/de-aggros

This seems to happen more regularly when you're fighting a dragon outside of the usual setting. It will attack a few times and then fly offscreen, disappearing into the distance and apparently losing hostility to the player.