Changes to the tests