diff --git a/browser_tests/assets/widgets/load_audio_widget.json b/browser_tests/assets/widgets/load_audio_widget.json
new file mode 100644
index 000000000..c40440a1c
--- /dev/null
+++ b/browser_tests/assets/widgets/load_audio_widget.json
@@ -0,0 +1,32 @@
+{
+  "last_node_id": 1,
+  "last_link_id": 0,
+  "nodes": [
+    {
+      "id": 1,
+      "type": "LoadAudio",
+      "pos": [41.5296516418457, 16.930862426757812],
+      "size": [315, 82],
+      "flags": {},
+      "order": 0,
+      "mode": 0,
+      "inputs": [],
+      "outputs": [
+        {
+          "name": "AUDIO",
+          "type": "AUDIO",
+          "links": null
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "LoadAudio"
+      },
+      "widgets_values": [null, ""]
+    }
+  ],
+  "links": [],
+  "groups": [],
+  "config": {},
+  "extra": {},
+  "version": 0.4
+}
diff --git a/browser_tests/widget.spec.ts b/browser_tests/widget.spec.ts
index 8ae545c85..099d2230b 100644
--- a/browser_tests/widget.spec.ts
+++ b/browser_tests/widget.spec.ts
@@ -134,3 +134,10 @@ test.describe('Load image widget', () => {
     await expect(comfyPage.canvas).toHaveScreenshot('load_image_widget.png')
   })
 })
+
+test.describe('Load audio widget', () => {
+  test('Can load audio', async ({ comfyPage }) => {
+    await comfyPage.loadWorkflow('widgets/load_audio_widget')
+    await expect(comfyPage.canvas).toHaveScreenshot('load_audio_widget.png')
+  })
+})
diff --git a/browser_tests/widget.spec.ts-snapshots/load-audio-widget-chromium-2x-linux.png b/browser_tests/widget.spec.ts-snapshots/load-audio-widget-chromium-2x-linux.png
new file mode 100644
index 000000000..4768e1024
Binary files /dev/null and b/browser_tests/widget.spec.ts-snapshots/load-audio-widget-chromium-2x-linux.png differ
diff --git a/browser_tests/widget.spec.ts-snapshots/load-audio-widget-chromium-linux.png b/browser_tests/widget.spec.ts-snapshots/load-audio-widget-chromium-linux.png
new file mode 100644
index 000000000..8f9584fe0
Binary files /dev/null and b/browser_tests/widget.spec.ts-snapshots/load-audio-widget-chromium-linux.png differ
diff --git a/src/services/litegraphService.ts b/src/services/litegraphService.ts
index 2fcdc91f9..822319336 100644
--- a/src/services/litegraphService.ts
+++ b/src/services/litegraphService.ts
@@ -170,7 +170,9 @@ export const useLitegraphService = () => {
     addNodeContextMenuHandler(node)
     addDrawBackgroundHandler(node)
     addNodeKeyHandler(node)
-
+    // Note: Some extensions expects node.comfyClass to be set in
+    // `beforeRegisterNodeDef`.
+    node.prototype.comfyClass = nodeDefV1.name
     await extensionService.invokeExtensionsAsync(
       'beforeRegisterNodeDef',
       node,
@@ -179,7 +181,6 @@ export const useLitegraphService = () => {
 
     const nodeDef = new ComfyNodeDefImpl(nodeDefV1)
     node.comfyClass = nodeDef.name
-    node.prototype.comfyClass = nodeDef.name
     node.nodeData = nodeDef
     LiteGraph.registerNodeType(nodeId, node)
     // Note: Do not following assignments before `LiteGraph.registerNodeType`