fix the unit tests