Modify the way to store the context.