Put the density in the eddy diffusion