Cosmetic changes to the unit test