Update jinja_context.py