How can i disable jump while swimming

Not working is related to animation, once i disable the IsInAir? boolean is doesnt jump on beeing at surface of water or exit from water