How can I switch back to walking mode from flying?

[Yarboa]

I made an attack that is in the air so when I press the right-click button it’s making the attack in the air basically.
and I set it to flying mode so I don’t have gravity issues and its working pretty well, but how can I set it up so that when she is ending the attack the character is going back to walking mode?
because when the character ends the attack she is flying.

[KryogenicGames]

what is your BP look like when it ends attack?

[Yarboa]

This is how I set it up, idk if it’s good or not.

[KryogenicGames]

try to set flying befor montage and when montage is done set movement mode back to walking if that works

[Yarboa]

So I have done something and I think its working.
I put a notify in the montage on when the character is falling, and in the BP I set the character’s movement mode to walking on the notify begin.

[KryogenicGames]

well done