Fix test following previous commit.