diff --git a/laborious/workflows/sub_workflows/format_and_export_prediction.py b/laborious/workflows/sub_workflows/format_and_export_prediction.py
index c9127c5..a9cda2a 100644
--- a/laborious/workflows/sub_workflows/format_and_export_prediction.py
+++ b/laborious/workflows/sub_workflows/format_and_export_prediction.py
@@ -198,6 +198,8 @@ class FormatAndExportPrediction:
                 'table_name': input_data['table_name'],
                 'data': prediction,
                 'timestamp_conversion': {'column': 'timestamp', 'format': DATETIME_FORMAT_WITH_TZ},
+                'on_conflict': input_data.get('on_conflict', 'replace'),
+                'unique_columns': ['model_id', 'timestamp'],
             },
             retry_policy=retry_policy,
             start_to_close_timeout=timedelta(seconds=180),