diff --git a/py/smZ_cfg_denoiser.py b/py/smZ_cfg_denoiser.py
index 7d46ca7..c25e123 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, KSamplerX0Inpaint, wrap_model
+from comfy.samplers import KSampler, KSamplerX0Inpaint
 #from comfy.k_diffusion.external import CompVisDenoiser
 from comfy.k_diffusion import sampling as k_diffusion_sampling
 from comfy import samplers