Add tests for schema