Fix the generator argument.