+fix stratified sampler