Fixed failed unittest.