fix UT failure cases