Woops, forgot to update the tests