Thanks for trying it out!
This was a tricky one, because I wanted to give the player control in the air so they can stick their landings, but yeah I also missed the momentum build up. I think I've found something in-between here that lets the player take control in the air to a lesser extent, but still preserves momentum in most cases.