I added the segmentation_col to the set of columns_used

and the test needed to account for this.  I think this is the desired behavior since the segmentation column needs to be added to the dataframe