Flying Animation Change

HI All

Just watched this Tutorial :Unreal Engine 4 - Improved Double Jump with Animations - YouTube

Now I have mostly adapted it to Side Scroller to the level I want except, when he comes out of flying he goes directly into the run mode.

I tried this with the Anim Graph, not luck.

It checks if Flying, if so, goes into Jump Loop, but when I release “W” he starts running while he is falling, Not sure why it is any different from when falling from a jump of off a cliff to when you stop flying.

I am guessing it is an easy fix, any help would be great,