fix bug on restored extra energy