Fixed a change in the API for the test classes.