diff --git a/comfy_extras/nodes_pag.py b/comfy_extras/nodes_pag.py
index c48a3958..63f43fd6 100644
--- a/comfy_extras/nodes_pag.py
+++ b/comfy_extras/nodes_pag.py
@@ -47,7 +47,7 @@ class PerturbedAttentionGuidance:
 
             return cfg_result + (cond_pred - pag) * scale
 
-        m.set_model_sampler_post_cfg_function(post_cfg_function, disable_cfg1_optimization=True)
+        m.set_model_sampler_post_cfg_function(post_cfg_function)
 
         return (m,)