[Misc] Fix failing test.

(cherry picked from commit 014ad3c77c689befb02d10445f9bd19aaad8ad75)