diff --git a/app/service/generate_batch_image/service.py b/app/service/generate_batch_image/service.py
index 2279382..6d007c8 100644
--- a/app/service/generate_batch_image/service.py
+++ b/app/service/generate_batch_image/service.py
@@ -6,15 +6,15 @@ from app.service.generate_batch_image.service_batch_pose_transform import batch_
 async def start_product_batch_generate(data):
     generate_clothes_task = batch_generate_product.delay(data.dict())
     print(generate_clothes_task)
-    product_publish_status(data.tasks_id, f"0/{data.batch_size}", "")
-    return {"task_id": data.tasks_id, "state": generate_clothes_task.state}
+    product_publish_status(data.batch_tasks_id, f"0/{len(data.batch_data_list)}", "")
+    return {"task_id": data.batch_tasks_id, "state": generate_clothes_task.state}
 
 
 async def start_relight_batch_generate(data):
     generate_clothes_task = batch_generate_relight.delay(data.dict())
     print(generate_clothes_task)
-    relight_publish_status(data.tasks_id, f"0/{data.batch_size}", "")
-    return {"task_id": data.tasks_id, "state": generate_clothes_task.state}
+    relight_publish_status(data.batch_tasks_id, f"0/{len(data.batch_data_list)}", "")
+    return {"task_id": data.batch_tasks_id, "state": generate_clothes_task.state}
 
 
 async def start_pose_transform_batch_generate(data):
diff --git a/app/service/generate_batch_image/service_batch_generate_relight_image.py b/app/service/generate_batch_image/service_batch_generate_relight_image.py
index d75b0a7..0a90646 100644
--- a/app/service/generate_batch_image/service_batch_generate_relight_image.py
+++ b/app/service/generate_batch_image/service_batch_generate_relight_image.py
@@ -184,5 +184,25 @@ if __name__ == '__main__':
             )
         ]
     )
-
     batch_generate_relight(rd.dict())
+    # X = {
+    #     "batch_tasks_id": "abcd",
+    #     "user_id": "89",
+    #     "batch_data_list": [
+    #         {
+    #             "tasks_id": "123-5464",
+    #             "product_type": "overall",
+    #             "image_url": "aida-users/89/product_image/02894523-19b5-46eb-a9c6-2f512f5fec84-0-89.png",
+    #             "prompt": "Colorful black",
+    #             "direction": "Right Light",
+    #         },
+    #         {
+    #             "tasks_id": "123-5464",
+    #             "product_type": "overall",
+    #             "image_url": "aida-users/89/product_image/02894523-19b5-46eb-a9c6-2f512f5fec84-0-89.png",
+    #             "prompt": "Colorful black",
+    #             "direction": "Right Light",
+    #         }
+    #
+    #     ]
+    # }