Oops, fix tests.