improve unittest.