[misc] Fix test.

(cherry picked from commit da947509dd9ea105d24869ffa7e2b4d72bfcf168)