Allay's head does not rotate to where it looks

The allay looks up and down as well as side to side but its head doesn't rotate up and down visually. It can also lag behind in horizontal rotation in some situations.

I presume it's because the head of the allay isn't its own model part, but rather part of the body, and uses body rotation instead of head rotation, but I haven't looked at the code.