fix UnitTest Error