correct implementation of the diffusion approximation