update the tests to the new architecture