Reset the energy if needed

Signed-off-by: ShadowChild <shadychild12@gmail.com>