fix unittest failure

fix unittest failure