Wait for standup animation to finish - speed variable

Hi all,

I am trying to use a Prone to Stand Animation but the speed variable for running, starts before the animation state is finished getting up so the character is moving at a fast speed before actually entering a running animation.

On top of this I am using Enum to change the state - Prone, Directional Movement, Crouch etc. So in the Character BP for the prone input im changing the state in the enum list. Screenshot below.
So i don’t know where to control the speed variable for movement before i want it to kick-in.
Should i use an anim montage?