Small fix for schemas