Updated the tests to reflect the new signature changes.