This is absolutely possible. It would be helpful to get screen grabs and/or your project first to see what you’re currently doing.
Jumping ahead of myself, one thought is that you could create an animation notify event that gets fired when the Locomotion state is fully blended in. Then, in your anim event graph, you’d catch that event and only then would you allow your WASD bool flag to be reset.
But there may be a simpler solution - just need to see what you’re doing first.