move that to the base so all tree models get it