Oops, forgot to update the tests.