add energy correction. See the last term of eq.33 JCP 139, 174107