diff --git a/js/external_tooling_nodes.js b/js/external_tooling_nodes.js
index 31e4a79..bf1747a 100644
--- a/js/external_tooling_nodes.js
+++ b/js/external_tooling_nodes.js
@@ -185,7 +185,7 @@ app.registerExtension({
         if (nodeData.name === "ETN_KritaCanvas") {
             setIconImage(nodeType, canvasIcon, [200, 100], 0, 2)
         } else if (nodeData.name === "ETN_KritaOutput") {
-            setIconImage(nodeType, outputIcon, [200, 120], 2, 0)
+            setIconImage(nodeType, outputIcon, [200, 100], 1, 0)
         } else if (nodeData.name === "ETN_Parameter") {
             setupParameterNode(nodeType)
         } else if (nodeData.name === "ETN_SendText") {
diff --git a/krita.py b/krita.py
index 813d6f7..d1b2434 100644
--- a/krita.py
+++ b/krita.py
@@ -74,12 +74,19 @@ AnyType = _AnyType("*")
 BasicTypes = _BasicTypes("BASIC")
 
 
-class KritaOutput(SendImageWebSocket):
+class KritaOutput:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {"required": {"images": ("IMAGE",)}}
+
     RETURN_TYPES = ()
     FUNCTION = "send_images"
     OUTPUT_NODE = True
     CATEGORY = "krita"
 
+    def send_images(self, images):
+        return SendImageWebSocket().send_images(images, "PNG")
+
 
 class KritaSendText:
     @classmethod