diff --git a/src/main/java/com/ai/da/common/RabbitMQ/GenerateConsumer.java b/src/main/java/com/ai/da/common/RabbitMQ/GenerateConsumer.java
index 3e8e6a6e..4e576307 100644
--- a/src/main/java/com/ai/da/common/RabbitMQ/GenerateConsumer.java
+++ b/src/main/java/com/ai/da/common/RabbitMQ/GenerateConsumer.java
@@ -116,7 +116,7 @@ public class GenerateConsumer {
         try {
             log.info("tasks_id : {} start ", generateResult.get("tasks_id"));
             if (generateResult.get("status").equals("SUCCESS")) {
-                String url = generateResult.get("image_url");
+                String url = generateResult.get("data");
                 String taskId = generateResult.get("tasks_id");
                 String category = generateResult.get("category");
                 generateService.processGenerateResult(taskId, url, category);