Prefer returning the context in the model instead of using setContext