Modified calculation of mass diffusion fluxes and correction velocity