Fix unit Test case