you'll need to change these tests with the new implementation