Taking damage a second before hitting the ground, as you can see, normally if you hit the ground there will be particles on the block.   Sometime you can take damage while you’re in the air.

Steps to reproduce:

1. Wear Iron armour and Elytra.  Enchant the trident with Riptide 3
2. Dash into the air with the trident
3. Hit the ground