BLD-951: Fix flaky tests.