small fix in generate script