Restore TestCase for unittests