fix unittest

fix unittest cases