Fix the unittest again.