Fix broken unittest.