diff --git a/laborious/activities/gates.py b/laborious/activities/gates.py
index 0e8dc82..12fafdf 100644
--- a/laborious/activities/gates.py
+++ b/laborious/activities/gates.py
@@ -523,7 +523,7 @@ class Gates(MinioManager):
         Args:
             input_data (dict): Input data containing:
                 - data (dict[str, Any]): Raw prediction data to format
-                - last_timestamp (str): Last timestamp of the data
+                - timestamp (str): Timestamp of the data
                 - model_id (str): Unique identifier for the ML model
                 - prediction_confidence (float): Confidence score for the prediction
                 - prediction_store_policy (str): Storage policy in format 'type:value'
@@ -532,7 +532,7 @@ class Gates(MinioManager):
             dict: Formatted prediction data ready for storage and export
         """
         metadata = input_data['metadata']
-        last_timestamp = input_data['last_timestamp']
+        last_timestamp = input_data['timestamp']
         prediction_store_policy = input_data['prediction_store_policy']
         self.info('Formatting prediction...', metadata)
 
diff --git a/laborious/workflows/sub_workflows/format_and_export_prediction.py b/laborious/workflows/sub_workflows/format_and_export_prediction.py
index 1cd2be1..a9cda2a 100644
--- a/laborious/workflows/sub_workflows/format_and_export_prediction.py
+++ b/laborious/workflows/sub_workflows/format_and_export_prediction.py
@@ -101,7 +101,7 @@ class FormatAndExportPrediction:
                 {
                     **metadata,
                     'data': data,
-                    'last_timestamp': input_data['last_timestamp'],
+                    'timestamp': input_data['timestamp'],
                     'model_id': input_data['model_id'],
                     'prediction_confidence': prediction_confidence,
                     'prediction_store_policy': input_data['prediction_store_policy'],