Fixed TestCase class