fix UnitTest unstable issue