diff --git a/py/smZ_cfg_denoiser.py b/py/smZ_cfg_denoiser.py
index e6a4ce1..54e61c6 100644
--- a/py/smZ_cfg_denoiser.py
+++ b/py/smZ_cfg_denoiser.py
@@ -4,7 +4,7 @@ import torch
 from typing import List
 import comfy.sample
 from comfy import model_base, model_management
-from comfy.samplers import KSampler, CompVisVDenoiser, KSamplerX0Inpaint
+from comfy.samplers import KSampler, KSamplerX0Inpaint
 from comfy.k_diffusion.external import CompVisDenoiser
 import nodes
 import inspect
@@ -318,4 +318,4 @@ def set_model_k(self: KSampler):
 class SDKSampler(comfy.samplers.KSampler):
     def __init__(self, *args, **kwargs):
         super(SDKSampler, self).__init__(*args, **kwargs)
-        set_model_k(self)
\ No newline at end of file
+        set_model_k(self)