When the riptide animation stops while swimming, you get the 1.8 block hitbox even though you are swimming

The bug

If you use in water a riptide trident and then swim with the swimming animation before the riptide animation stops you get the 1.8 blocks high hitbox after the animation stops. Which means you cannot swimm through 1*1 gaps. 

How to reproduce:

1. Take a riptide trident

2. Go in water for example in an ocean

3. Use the riptide trident 

4. Swim with the swimming animation before the riptide animation stops 

5. Swim up against a block

What I observed: You stop moving upwards at a lower position than you should, which means that you have the 1.8 block hitbox which also means that you can’t swim through 1*1 gaps.

What I expected: When the riptide animation stops you still have the smaller hitbox because you are swimming with the swimming animation. 

That seems to happen because the game doesn’t test if you do something were you should have the 1 block hitbox after the animation stops, simelar to MCPE-55862 and MCPE-54255.