Fix the problem with the eddy diffusion by proper N2 calculation