fixed stochastic_sampler output