diff --git a/src/utils/agents/persona.py b/src/utils/agents/persona.py
index 1bd11ed..b14c311 100644
--- a/src/utils/agents/persona.py
+++ b/src/utils/agents/persona.py
@@ -324,15 +324,18 @@ class PersonaGenerator(Agent):
             message.response = prompt
             yield message
 
+            logger.info("Beginning image generation...")
             request = ImageRequest(filepath=os.path.join(defines.user_dir, persona["username"], f"profile.png"), prompt=prompt)
             placeholder = Message(prompt=prompt)
             async for placeholder in generate_image_status(
                 message=placeholder, 
                 **request.model_dump()
             ):
+                logger.info("Image generation continue...")
                 if placeholder.status != "done":
                     placeholder.response = placeholder.response
-                    yield message
+                    yield placeholder
+            logger.info("Image generation done...")
             persona["has_profile"] = True
 
             #