Modify TestCaseGenerator