Mace does not reset fall damage when damaging while flying with an elytra

When flying using an elytra, you can dive into an entity and hit with a mace to get the extra damage. However, this will not reset your fall damage and will most often cause the player to die.

To reproduce:
Fly in the air with an elytra, then dive to an entity and attack it with the mace. The mace will receive extra damage, but the fall damage will not be reset.