diff --git a/__pycache__/efficiency_nodes.cpython-312.pyc b/__pycache__/efficiency_nodes.cpython-312.pyc
new file mode 100644
index 0000000..19ca335
Binary files /dev/null and b/__pycache__/efficiency_nodes.cpython-312.pyc differ
diff --git a/efficiency_nodes.py b/efficiency_nodes.py
index 286d84d..03677b8 100644
--- a/efficiency_nodes.py
+++ b/efficiency_nodes.py
@@ -506,7 +506,7 @@ class TSC_KSampler:
                     rng_source, cfg_denoiser, add_seed_noise, m_seed, m_weight = script["noise"]
                     smZ_rng_source.rng_rand_source(rng_source) # this function monkey patches comfy.sample.prepare_noise
                     if cfg_denoiser:
-                        comfy.samplers.KSampler = smZ_cfg_denoiser.SDKSampler
+                        smZ_cfg_denoiser.register_hooks()
                     if add_seed_noise:
                         comfy.sample.prepare_noise = cg_mixed_seed_noise.get_mixed_noise_function(comfy.sample.prepare_noise, m_seed, m_weight)
                     else:
diff --git a/py/__pycache__/__init__.cpython-312.pyc b/py/__pycache__/__init__.cpython-312.pyc
new file mode 100644
index 0000000..a30962e
Binary files /dev/null and b/py/__pycache__/__init__.cpython-312.pyc differ
diff --git a/py/__pycache__/smZ_cfg_denoiser.cpython-312.pyc b/py/__pycache__/smZ_cfg_denoiser.cpython-312.pyc
new file mode 100644
index 0000000..da1d6fd
Binary files /dev/null and b/py/__pycache__/smZ_cfg_denoiser.cpython-312.pyc differ