diff --git a/laborious/workflows/sub_workflows/prediction_process.py b/laborious/workflows/sub_workflows/prediction_process.py
index e4fc8cc..1e01f6a 100644
--- a/laborious/workflows/sub_workflows/prediction_process.py
+++ b/laborious/workflows/sub_workflows/prediction_process.py
@@ -205,7 +205,7 @@ class PredictionProcess():
                 {
                     'schema': schema,
                     'table_name': table_name,
-                    'model_id': model_id
+                    'model': model_id
                 },
                 retry_policy=retry_policy,
                 start_to_close_timeout=timedelta(minutes=1),