For anyone looking at this wondering how I got it working, I did it using the Time Remaining node in the animbp state machine.
https://docs.unrealengine.com/en-us/Gameplay/HowTo/CharacterMovement/Blueprints/AnimBlueprint_Jump
After that I ditched the Play Animation functions in my event graph and instead made a couple variables to trigger the state change. Hope this helps somebody