Modify unit tests to account for new changes