minor unit test improvement