If we hit a fish and it comes out of the water, it rising to upward.

(While writing this text, I am using translate, so there may be mistakes)
if after hitting a fish, it come out of the water, fish starts to rise up after falling down, it drown and dies if it doesn't go back to the water.

Expected Result:
When fish come out of the water and falling down , it should die.

Actual Result:
after fish getting out of water, it doesn't fall down and die.

 How to reproduce:
1.build a 3x2x3 inside empty platform.
2.place a block on one side of the platform that the fish can pass through but block the water.
3.Spawn fish and hit while it come out of the water.
4. if the fish comes out of the water, at first it will go down a little but then it will start to rise.