Update the tests to reflect recent code changes