Little fix in generator