swimming suffocation

When you swim and you stop/position upwards, if you have a block on your head you suffocate inside that block or even deeper even though you have a two or more space where you where swimming.

Also this happens when swimming in a one block high long tunnel, and like with the elytra in the Java version, if you are in a one block space tunnel and you top sprinting, you should still stay in the swimming position.