Sorting by other field (when clicking header) in ExpressGrid 6

Hi,
I got this code from you for drawing the header yourself in ExpressGrid 5 and this seems necessary still in the new ExpressGrid 6 (if you want to sort by another field than the one shown in the grid). This old code does not show the new header highlighting correctly though. Do you have some new code for this?
Here is the old code:
procedure TViewManager_frm.grid_ViewEntryDBBandedTableView1TitleCustomDrawHeader(
  Sender: TcxGridTableView; ACanvas: TcxCanvas;
  AViewInfo: TcxGridColumnHeaderViewInfo; var ADone: Boolean);
var
  I: Integer;
  R: TRect;
begin
  with AViewInfo do
  begin
    Sender.Painter.LookAndFeelPainter.DrawHeader(ACanvas, Bounds, TextAreaBounds,
    Neighbors, Borders, cxbsNormal, taLeftJustify, vaTop, False, False, Column.Caption, ACanvas.Font, clNone, Color);
    with TcxGridColumnHeaderViewInfoAccess(AViewinfo) do
      for I := 0 to AreaViewInfoCount - 1 do
      begin
        if AreaViewInfos[I] is TcxGridColumnHeaderFilterButtonViewInfo then
           Sender.Painter.LookAndFeelPainter.DrawFilterDropDownButton(Canvas, AreaViewInfos[I].Bounds,
           GridCellStateToButtonState(AreaViewInfos[I].State), TcxGridColumnHeaderFilterButtonViewInfo(AreaViewInfos[I]).Active);
        R := AreaViewInfos[I].Bounds;
        R.Right := AreaViewInfos[I].Bounds.Left - 2;
        R.Left := R.Right - Sender.Painter.LookAndFeelPainter.SortingMarkSize.X;
        R.Bottom := R.Bottom - 4;
        R.Top := R.Top + 4;
      end;
    if grid_ViewEntryDBBandedTableView1TitleSort.SortOrder <> soNone then
    begin
      ACanvas.Brush.Color := AViewInfo.Params.Color;
      InflateRect(R, 1, 1);
      ACanvas.FillRect®;
      InflateRect(R, -1, -1);
      Sender.Painter.LookAndFeelPainter.DrawSortingMark(ACanvas, R, grid_ViewEntryDBBandedTableView1TitleSort.SortOrder = soAscending);
    end;
    ADone := True;
  end;
end;

Best Regards,
Johan