Another little fix to the traditional method diag.