Minor cleanup in test case.