Small fix to schema