Changed diffusion tests to be (slightly) more fair