Fixed a flaky test