Fix test failure from previous change.