BatchSampling bugfixes