Small fix in the generated model