Fix for volume_integral diagnostics