Correcting kinetic energy computation