Add missing line of code (incomplete energy summation).