diff --git a/orchestrator/utils/orchestrator_functions.py b/orchestrator/utils/orchestrator_functions.py
index ac50f32..b8e7eec 100644
--- a/orchestrator/utils/orchestrator_functions.py
+++ b/orchestrator/utils/orchestrator_functions.py
@@ -248,6 +248,8 @@ def predictions_batch(config: dict[str, Any]):
         'datetime_columns': config.get('datetime_columns', []),
         'schema': 'sientia_data',
         'table_name': 'predictions',
+        'save_transform': config.get('save_transform', True),
+        'transform_table_name': 'transformed_data',
         'retention_time': config.get('model_retention_minutes', 60) * 60,
         'opc_output_config': tags,
         'input_filters': overlap_filter_config(
diff --git a/tests/orchestrator/utils/test_orchestrator_functions.py b/tests/orchestrator/utils/test_orchestrator_functions.py
index 4dc9666..4db42a8 100644
--- a/tests/orchestrator/utils/test_orchestrator_functions.py
+++ b/tests/orchestrator/utils/test_orchestrator_functions.py
@@ -238,6 +238,8 @@ def test_predictions_batch(mock_process_path_priority, mock_overlap_filter_confi
         'query': 'test_query',
         'schema': 'sientia_data',
         'table_name': 'predictions',
+        'save_transform': True,
+        'transform_table_name': 'transformed_data',
         'retention_time': 60 * 60,
         'opc_output_config': {
             'test_server_id': {