tsan: fix flaky test