diff --git a/model_server/util.py b/model_server/util.py
index 8bc071ab056570a594658a0be4fa99eeb9c875dd..1cc33753dbe477e3ddff686f2a5459a93386b7bb 100644
--- a/model_server/util.py
+++ b/model_server/util.py
@@ -1,10 +1,12 @@
 from pathlib import Path
 import re
 from time import localtime, strftime
+from typing import List
 
 import pandas as pd
 
 from model_server.accessors import InMemoryDataAccessor, write_accessor_data_to_file
+from model_server.models import Model
 
 def autonumber_new_directory(where: str, prefix: str) -> str:
     """
@@ -75,6 +77,7 @@ def loop_workflow(
         files: list,
         output_folder_path: str,
         workflow_func: callable,
+        models: List[Model],
         params: dict,
         export_batch_csvs: bool = True,
         write_intermediate_products: bool = True,
@@ -95,6 +98,7 @@ def loop_workflow(
         export_kwargs = {
             'input_file_path': ff,
             'output_folder_path': output_folder_path,
+            'models': models,
             **params,
         }