Fixed a flaky test.

Review: https://reviews.apache.org/r/23543