Small cleanup in unit test.