Add diag upper checking