Whoops, forgot to update a test.