fixed some unit test