Fixing flaky test.