make flaky test more robust.