Fix class for test case.