Correct one unit test, to reflect new  function