Update the tests, too.