Fix unittest.case import