TEZ-1703. addendum - fix flaky test. (zjffdu)

(cherry picked from commit 8f8a81f7a17f9018ae4e87bf0fca9d6cdc0a5ba4)