Add missing import for schema validation