try to improve flaky test.