Fix energy wobble