Small fix for missing schema