Updated tests to match the changed API.