Modify generator schema