update the test to make it more stable on multiple runs