Fix TestCase class