Passing tree along with add_row_via_wrapper seems to work.