actually fix the unittest...