Fix flaky time sensitive spec