Fix a couple of minor issues in the CMB stress test