unit test fixes.