Trivial clean up of unit test.