From f42f650b4c13f2dd1419fd4ade81527388153da2 Mon Sep 17 00:00:00 2001
From: JohnWL <34081873+John-WL@users.noreply.github.com>
Date: Tue, 9 Jan 2024 16:02:50 -0500
Subject: [PATCH] rename

---
 lib_comfyui/custom_extension_injector.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib_comfyui/custom_extension_injector.py b/lib_comfyui/custom_extension_injector.py
index 92761d5..44ed24b 100644
--- a/lib_comfyui/custom_extension_injector.py
+++ b/lib_comfyui/custom_extension_injector.py
@@ -102,8 +102,8 @@ def add_server__init__patch(callback):
     import server
     original_init = server.PromptServer.__init__
 
-    def patched_PromptQueue__init__(*args, **kwargs):
+    def patched_PromptServer__init__(*args, **kwargs):
         callback(*args, **kwargs)
         original_init(*args, **kwargs)
 
-    server.PromptServer.__init__ = patched_PromptQueue__init__
+    server.PromptServer.__init__ = patched_PromptServer__init__