fix unittest failure