Fixed the flaky test.