Thanks, you helped me figure out what was causing this for me.
I know its a couple years late but heres how i just fixed it
I have a first state of swimming which changes the buoyancy from 1 to 1.5 when i enter to keep the player at the top of the water. So i used that value as a check as to whether the jump animation should trigger. you could use something like IsSwimming as a bool as your check in a similar way