Minor clean up within test.