Fixed another flaky test