Thank you for playing!
Yes that it's because i forced that if you jump out of the water too quick multiple times it cancels your momentum (At first you could hold your mouse up and keep jumping forever)
I'm sure there was a more elegant solution for it that i couldn't figure out in time haha
Endless mode would be neat you're right!