fix the unitest to meet the new function declaration.