Update the tests