Fixes a flaky test