Hopefully fixed the flaky test