Flaky spec fix