fix binning with stratification