Remove unused import
Fix bug in EnergyDrain