fixed flaky test