Cleaning up unit tests.