Unit test fixes.