Fixed kinetic energy bug