Fix new unittest