Fix failing test.
(cherry picked from commit 261f85491afc49935cde4df19718cadb983a6633)