minor changes to generate