Fixed broken HMC sampler