diff --git a/laborious/activities/gates.py b/laborious/activities/gates.py
index 75d4b93..1bf37f2 100644
--- a/laborious/activities/gates.py
+++ b/laborious/activities/gates.py
@@ -512,7 +512,7 @@ class Gates(MinioManager):
         prediction_store_policy = input_data['prediction_store_policy']
         self.info('Formatting prediction...', metadata)
 
-        payload: MinioDataFramePayload = input_data['data']
+        payload = MinioDataFramePayload.from_dict(input_data['data'])
         data = await payload.retrieve(self.minio_repository, metadata)
 
         # Create timestamp column from index and reset index