Changed argument order in kernel call in diffusion