More fixes to test_mcmc.py