forgot to update the tests