borked attempt at a solution

common/src/main/kotlin/com/cobblemon/mod/common/api/riding/behaviour/RidingBehaviour.kt

@@ -105,7 +105,7 @@ interface RidingBehaviour<Settings : RidingBehaviourSettings, State : RidingBeha
 
 
     /**
-     * Internal helpers to help behaviour specific calculations
+     * Internal helpers for behaviour specific calculations
      */
     companion object {
         /**

common/src/main/kotlin/com/cobblemon/mod/common/api/riding/behaviour/RidingController.kt

@@ -111,6 +111,7 @@ class RidingController(
             // check if style is water and player is submerged
             val isEyeInLiquid = entity.isEyeInFluid(FluidTags.WATER) || entity.isEyeInFluid(FluidTags.LAVA)
             if (context?.style == RidingStyle.LIQUID && isEyeInLiquid) return false
+            if (!entity.onGround() && driver.jumping) return true
             return (driver as DoubleJump).isDoubleJumping
         }
         else {