Fix output of diag command