Fixed the test case generation