fixed up unit tests.