Use jinja_env instead of injected context