Cleaning up unit test.