Try to fix flaky test