Fix flaky spec