fixed up unit test.