a little tweak to test/sampling.jl