Update for dntdiag api change