My suggestion is for the airsafe option in safewalk to only work if the block you're jumping onto is bellow the Y value of the current block you're standing on, so that if it is on the same Y level or above, it won't stop you mid air and you can move freely