tiny bug in Diffusion