Any way to make custom mob act like a parrot?

Java 1.20.1, MCreator 2024.4.52410

So I'm making birds and I have animations of walk and flight. For walk I have playback condition Return logic | is event/target entity on ground, but when entity on the ground it's just continues to play flight animation until entity will walk. Is it possible to stop flight anim when entity touches and stays on the ground?