Correct the generator name bug