Fixed flaky test