Attempt to fix flakey unit test.