fix unittest again