Fix flaky (#883)