I play ascended and when I dismount any water creature it acts like I dismounted a land creature so you can’t swim. I don’t know how to fix it in survival mode because I always play in creative mode. (To fix this in creative mode you have to double tap the jump button. This might work in survival mode but I’m not sure.)