Fix validations

Fixes some validation problems in model_checks.py