Fixing a flaky test.