Whoops, forgot to update tests.