Fixed a flaky test.