add robustness to stochastic test