Fix diffusion equation.