fix another unit test