update the unittest