Working on energy conservation for diffuse term.