Update tests according to the previous commit.