Update tests to new changes.