Player freezes when sprint-swimming underneath a block

After hitting a mob while sprint-swimming, player freezes when sprint-swimming up >1 block away from the nearest overhead block.

What I expected to happen was...:
I expected to collide with the block above me as I swam up towards it.

What actually happened was...:
I teleported repeatedly into the space two blocks below the nearest overhead block. I call this "freezing."

Steps to Reproduce:
1. Sprint-swim

2. Hit a mob while sprint-swimming

3. Continue sprint-swimming

4. Ascend towards any solid block overhead while sprint-swimming