diff --git a/py/luma_nodes.py b/py/luma_nodes.py
index f41633f..98bbef6 100644
--- a/py/luma_nodes.py
+++ b/py/luma_nodes.py
@@ -10,7 +10,8 @@ from .lumaai_api_node import (
     CharacterReference,
     ImageGeneration,
     ModifyImage,
-    Ray2Text2Video,
+    AddAudio2Video,
+    UpscaleGeneration,
 )
 from .imgbb_node import ImgBBUpload
 
@@ -27,7 +28,8 @@ NODE_CLASS_MAPPINGS = {
     "CharacterReference": CharacterReference,
     "LumaImageGeneration": ImageGeneration,
     "LumaModifyImage": ModifyImage,
-    "Ray2Text2Video": Ray2Text2Video,
+    "LumaAddAudio2Video": AddAudio2Video,
+    "LumaUpscaleGeneration": UpscaleGeneration,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
@@ -43,5 +45,6 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "LumaCharacterReference": "Character Reference",
     "LumaImageGeneration": "Image Generation",
     "LumaModifyImage": "Modify Image",
-    "Ray2Text2Video": "Ray2 Text to Video",
+    "LumaAddAudio2Video": "Add Audio to Video",
+    "LumaUpscaleGeneration": "Upscale Generation",
 }