Improve TestCase