updated to point to diag