diff --git a/__init__.py b/__init__.py
index 83fa53c..c90967b 100644
--- a/__init__.py
+++ b/__init__.py
@@ -32,10 +32,16 @@ original_validate = execution.validate_prompt
 
 
 @wraps(original_validate)
-async def hijack_validate(prompt_id: int, prompt: dict) -> bool:
+async def hijack_validate(*args):
+    for arg in args:
+        if isinstance(arg, dict):
+            prompt = arg
+            break
+
     if not find_node(prompt):
         disable_recenter()
-    return await original_validate(prompt_id, prompt)
+
+    return await original_validate(*args)
 
 
 execution.validate_prompt = hijack_validate
diff --git a/pyproject.toml b/pyproject.toml
index 77127f9..1162254 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-diffusion-cg"
 description = "Color Grading for Stable Diffusion"
-version = "1.2.0"
+version = "1.3.0"
 license = { text = "MIT License" }
 dependencies = []