ENH: use set_data