Minor unit test improvement