Correcting k(grad U) expression