Fix rthe unittest