Tweak stochastic tests