Fix jsonschema mode
Add more tests