small changes to UnitTest
minor fix for UnitTest