fix unit test
cosmetic changes