Fix test failures #7
Description
Recent updates to the code seem to have caused most of the tests to fail. They should be updated to reflect these changes.
{{ message }}
Recent updates to the code seem to have caused most of the tests to fail. They should be updated to reflect these changes.