fix presumed bug in energy calc