diff --git a/custom_routes.py b/custom_routes.py
index 10ef50f..4eb1563 100644
--- a/custom_routes.py
+++ b/custom_routes.py
@@ -1226,10 +1226,9 @@ async def send_json_override(self, event, data, sid=None):
                 "node_id": node,
                 "node_class": class_type,
             }
-            # if class_type == "PreviewImage":
-            #     logger.info("Skipping preview image")
-            # else:
-            if class_type == "ComfyDeployStdOutputImage":
+            if class_type == "PreviewImage":
+                logger.info("Skipping preview image")
+            else:
                 await update_run_with_output(
                     prompt_id,
                     data.get("output"),