Minor cosmetic on unit tests