improve diag