fix up unittest.