Fixed unit test issue.