diff --git a/scripts/dynamic_thresholding.py b/scripts/dynamic_thresholding.py
index 6cf1fec..550289b 100644
--- a/scripts/dynamic_thresholding.py
+++ b/scripts/dynamic_thresholding.py
@@ -189,7 +189,7 @@ class CustomVanillaSDSampler(sd_samplers_compvis.VanillaStableDiffusionSampler):
 
 ######################### K-Diffusion Implementation logic #########################
 
-class CustomCFGDenoiser(sd_samplers_kdiffusion.CFGDenoiser):
+class CustomCFGDenoiser(sd_samplers_kdiffusion.CFGDenoiserKDiffusion):
     def __init__(self, model, dtData):
         super().__init__(model)
         self.main_class = dtData