diff --git a/nodes/__init__.py b/nodes/__init__.py
index 66bd10c..f43a6bf 100644
--- a/nodes/__init__.py
+++ b/nodes/__init__.py
@@ -209,11 +209,12 @@ class ARDrawSignal:
             }
         }
 
+    def main(self, signal):
+        img_tensor = draw_signal(signal['y'], signal['sr'])
+        return (img_tensor,)
+
+
 NODE_CLASS_MAPPINGS["ARDrawSignal"] = ARDrawSignal
 NODE_DISPLAY_NAME_MAPPINGS["ARDrawSignal"] = "Draw Audio Signal"
 
 ############################################
-
-    def main(self, signal):
-        img_tensor = draw_signal(signal['y'], signal['sr'])
-        return (img_tensor,)