diff --git a/app/schemas/generate_image.py b/app/schemas/generate_image.py
index 815dd41..3e150d5 100644
--- a/app/schemas/generate_image.py
+++ b/app/schemas/generate_image.py
@@ -35,7 +35,7 @@ class GenerateProductImageModel(BaseModel):
 class Flux2ToProductImgModel(BaseModel):
     tasks_id: str
     prompt: str
-    image_url: str
+    image_path: str
     image_strength: int