You need to update your tests.