Small fix to the data model