Updating the tests