Allow setting the context