Flying up causes player to play strafe right animation?

Hi, I 'm trying to implement a flying mechanic in game. I have 4 different animation for forward, backward, left and right (they are all set up in a blendspace). I also want to be able to fly up and down.

My issue is that when flying upwards (or downwards) the direction of the player is 90 (I calculated the direction with calculateDirection, using the rotation and Velocity) so the blendspace plays the animation which corresponds to direction = 90 which is the fly right animation. So I know why the bug happens, but I don’t know how to solve it, any ideas? I can post blueprints screenshot if you need them.

Thanks in advance.

P.S. to fly up/down I use the same code as moving forward but with up vector