Fix the diffusion time