this feels a bit hacky. `augment_row_with_cell` already gets the header and I don't think that variable should be changed in the implementation. This will cause problems in modules and that overwrite this function. In my opinion we should already pass in the right header.