Tweak flaky unit test - seems to pass consistently now