diff --git a/comfyui/nodes/pivoine_diffrhythm.py b/comfyui/nodes/pivoine_diffrhythm.py
index 420b5f6..829c223 100644
--- a/comfyui/nodes/pivoine_diffrhythm.py
+++ b/comfyui/nodes/pivoine_diffrhythm.py
@@ -9,18 +9,16 @@ Author: valknar@pivoine.art
 import sys
 sys.path.append('/workspace/ComfyUI/custom_nodes/ComfyUI_DiffRhythm')
 
-# Monkey-patch the infer function to force chunked=False
-import DiffRhythmNode
-_original_infer = DiffRhythmNode.infer
+# Monkey-patch decode_audio from infer_utils to force chunked=False
+import infer_utils
+_original_decode_audio = infer_utils.decode_audio
 
-def patched_infer(*args, **kwargs):
-    # Force chunked to False if present
-    if 'chunked' in kwargs:
-        kwargs['chunked'] = False
-    return _original_infer(*args, chunked=False, **kwargs)
+def patched_decode_audio(latent, vae_model, chunked=True):
+    """Patched version that always uses chunked=False"""
+    return _original_decode_audio(latent, vae_model, chunked=False)
 
 # Apply the monkey patch
-DiffRhythmNode.infer = patched_infer
+infer_utils.decode_audio = patched_decode_audio
 
 from DiffRhythmNode import DiffRhythmRun