diff --git a/nornir_jinja2/plugins/tasks/template_file.py b/nornir_jinja2/plugins/tasks/template_file.py
index f140ace..2005d52 100644
--- a/nornir_jinja2/plugins/tasks/template_file.py
+++ b/nornir_jinja2/plugins/tasks/template_file.py
@@ -33,6 +33,7 @@ def template_file(
 
     if jinja_env:
         env = jinja_env
+        env.loader = FileSystemLoader(path)
     else:
         env = Environment(
             loader=FileSystemLoader(path), undefined=StrictUndefined, trim_blocks=True,