Fix generator test.