fix unittest bugs