Fixed a bug with HMC and tracking the correct positions/gradient after a rejection.