Add tests for validations