Update DiagHelper.php