Fix array handling in SchemaOf type

- Move array before object so that inference works correctly
- Include lazy as part of array type, to match the class definition

src/index.ts

@@ -6,7 +6,7 @@ import DateSchema, { create as dateCreate } from './date';
 import ObjectSchema, { AnyObject, create as objectCreate } from './object';
 import ArraySchema, { create as arrayCreate } from './array';
 import { create as refCreate } from './Reference';
-import { create as lazyCreate } from './Lazy';
+import Lazy, { create as lazyCreate } from './Lazy';
 import ValidationError from './ValidationError';
 import reach from './util/reach';
 import isSchema from './util/isSchema';
@@ -37,11 +37,11 @@ function addMethod(schemaType: any, name: string, fn: any) {
   schemaType.prototype[name] = fn;
 }
 
-type SchemaOf<T> = T extends AnyObject
-  ? ObjectSchema<{ [k in keyof T]: SchemaOf<T[k]> }>
-  : T extends Array<infer E>
-  ? ArraySchema<SchemaOf<E>>
-  : BaseSchema<Maybe<T>, AnyObject, T>;
+type SchemaOf<T> = T extends Array<infer E>
+    ? ArraySchema<SchemaOf<E> | Lazy<SchemaOf<E>>>
+    : T extends AnyObject
+    ? ObjectSchema<{ [K in keyof T]: SchemaOf<T[K]> }>
+    : BaseSchema<Maybe<T>, AnyObject, T>;
 
 export type AnyObjectSchema = ObjectSchema<any, any, any, any>;
 export type { SchemaOf, TypeOf, Asserts, Asserts as InferType, AnySchema };