diff --git a/orchestrator/utils/orchestrator_functions.py b/orchestrator/utils/orchestrator_functions.py
index 38bde51..b49a771 100644
--- a/orchestrator/utils/orchestrator_functions.py
+++ b/orchestrator/utils/orchestrator_functions.py
@@ -48,6 +48,7 @@ def minimal_retrain(config: dict[str, Any]):
         "query": config['query'],
         "schema": "sientia_data",
         "table_name": "log_retrain",
+        "datetime_columns": config.get('datetime_columns', []),
     }
 
 
@@ -187,6 +188,7 @@ def predictions_batch(config: dict[str, Any]):
         **common_config(config),
 
         "query": config['query'],
+        "datetime_columns": config.get('datetime_columns', []),
         "schema": "sientia_data",
         "table_name": "predictions",
         "retention_time": config.get('model_retention_minutes', 60) * 60,
diff --git a/values.yaml b/values.yaml
index c2c69e9..300ea80 100644
--- a/values.yaml
+++ b/values.yaml
@@ -11,7 +11,7 @@ image:
   # This sets the pull policy for images.
   pullPolicy: Always
   # Overrides the image tag whose default is the chart appVersion.
-  tag: "0.4.2"
+  tag: "0.4.4"
 
 # This is for the secrets for pulling an image from a private repository more information can be found here: https://kubernetes.io/docs/tasks/configure-pod-container/pull-image-private-registry/
 imagePullSecrets: