Add domain collection to generated models
Validate model in test