Fix Unit Test issue.