fix unittest