diff --git a/services/training/app/ml/trainer.py b/services/training/app/ml/trainer.py
index 12b78fab..beba1017 100644
--- a/services/training/app/ml/trainer.py
+++ b/services/training/app/ml/trainer.py
@@ -452,7 +452,7 @@ class BakeryMLTrainer:
                     # Update products completed for accurate tracking
                     self.status_publisher.products_completed = completed_products
                     
-                    await self.status_publisher.product_completed(
+                    await self.status_publisher.progress_update(
                         progress=progress,
                         step="model_training",
                         current_product=product_name,