Mob with no-gravity getting pushed up into air, never falling down.

Hey,

first of all, this is not really a bug as it is the expected behaviour. Anyways, maybe anyone knows an alternative approach to solve the problem:

We have a mob that has NoGravity set to true, it does not have gravity as expected. Now, when shooting the mob with a bow with the enchantment Punch, the mob gets knocked back and gets pushed up into the air, but does not fall down again, with the next hit it goes even higher, etc.

Maybe anyone knows a way to solve this problem?

Greets