I hope i don't bother you.
But you should add this to the imports as well.
(I know this is NMS code but there is already an CraftBukkit import section so this won't cause any additional work and improves readability)
Don't forget the DamageCause part.