Updating all the tests