diff --git a/nodes.py b/nodes.py
index 1a13588..6cbdb96 100644
--- a/nodes.py
+++ b/nodes.py
@@ -40,8 +40,8 @@ comfy_graph = None
 comfy_graph_utils = None
 
 try:
-	import comfy.graph as temp_graph
-	import comfy.graph_utils as temp_graph_utils
+	import comfy_execution.graph as temp_graph
+	import comfy_execution.graph_utils as temp_graph_utils
 	comfy_graph = temp_graph
 	comfy_graph_utils = temp_graph_utils
 	print("\033[95m" + f"{lib0246.HEAD_LOG}Topological Execution is detected." + "\033[0m")
@@ -3202,4 +3202,4 @@ NODE_DISPLAY_NAME_MAPPINGS.update({
 	"0246.Tag": "Tag",
 })
 
-print("\033[95m" + lib0246.HEAD_LOG + "Loaded all nodes and apis." + "\033[0m")
\ No newline at end of file
+print("\033[95m" + lib0246.HEAD_LOG + "Loaded all nodes and apis." + "\033[0m")