Falling through a block of water sometimes does not reset fall distance/damage

H4. Steps to reproduce

1. Open MCPE-156578.mcworld
2. Step down onto the red nether brick.
3. Step into the chute.
4. Respawn
5. Step onto the diamond block.
6. Step into the chute.

Expected result

The water near the bottom of the chute slows you every time.

Observed result

The water near the bottom of the chute slows you when you step off the diamond block, but not when you step off the red nether brick block. (The reason why is explained in the comment below.)

---

First i walk in to me mine and i got water under it and it only 2 blocks high so i lost 30 levels so rip but can this bug be fix or