Fix the MCMC tests