From fd929ff834e8160b861fb50e287f42e1693cc4b6 Mon Sep 17 00:00:00 2001
From: David Marx <david.marx84@gmail.com>
Date: Mon, 1 Jan 2024 18:34:16 -0800
Subject: [PATCH] Update __init__.py

---
 nodes/__init__.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/nodes/__init__.py b/nodes/__init__.py
index 8adfeb9..58be985 100644
--- a/nodes/__init__.py
+++ b/nodes/__init__.py
@@ -220,6 +220,12 @@ class ARDrawSignal:
         return (img_tensor,)
 
 
+NODE_CLASS_MAPPINGS["ARDrawSignal"] = ARDrawSignal
+NODE_DISPLAY_NAME_MAPPINGS["ARDrawSignal"] = "Draw Audio Signal"
+
+############################################
+
+
 class SignalToCurve:
     CATEGORY = f"{CATEGORY}"
     FUNCTION = "main"
@@ -244,7 +250,7 @@ class SignalToCurve:
         driving_signal_kf = kf.Curve({t:v for t,v in zip(frame_time, driving_signal)}, label=signal_name)
         return (driving_signal_kf,)
 
-NODE_CLASS_MAPPINGS["ARDrawSignal"] = ARDrawSignal
-NODE_DISPLAY_NAME_MAPPINGS["ARDrawSignal"] = "Draw Audio Signal"
+NODE_CLASS_MAPPINGS["SignalToCurve"] = SignalToCurve
+NODE_DISPLAY_NAME_MAPPINGS["SignalToCurve"] = "SignalToCurve"
 
 ############################################