Add fix for latest zod version