Add support for setting the `context_json` on set instances