Hopefully fix flakey unit test.