Fixed minor unit test issue.