diff --git a/src/deforum/generators/comfy_utils.py b/src/deforum/generators/comfy_utils.py
index 85f20ad..b0dd65a 100644
--- a/src/deforum/generators/comfy_utils.py
+++ b/src/deforum/generators/comfy_utils.py
@@ -166,7 +166,7 @@ def ensure_comfy(custom_path=None):
 
         import asyncio
         import execution
-        from nodes import init_custom_nodes
+        from nodes import init_extra_nodes as init_custom_nodes
         import server
 
         # Creating a new event loop and setting it as the default loop