Unit test bug fixes.