Clean up unit test.