Correct energy upon landing on the track