modify test code (UnitTestSample)