Fix test failure that was introduced by previous commit.