diff --git a/backend/open_webui/utils/images/comfyui.py b/backend/open_webui/utils/images/comfyui.py
index 11a0ebb8fe..6447a69f88 100644
--- a/backend/open_webui/utils/images/comfyui.py
+++ b/backend/open_webui/utils/images/comfyui.py
@@ -91,6 +91,7 @@ async def _ws_get_images(ws, workflow, client_id, base_url, api_key):
         node_output = history['outputs'][node_id]
         if node_id in workflow and workflow[node_id].get('class_type') in [
             'SaveImage',
+            'SaveImageAdvanced',
             'PreviewImage',
         ]:
             if 'images' in node_output: