Fixed some generator test.