Fix unittest.