diff --git a/__init__.py b/__init__.py
index 9092103..f7f19a5 100644
--- a/__init__.py
+++ b/__init__.py
@@ -94,12 +94,12 @@ class CADS:
             c = args["c"]
 
             if noise_scale > 0.0:
-                apply_to = c.get(key, c["c_crossattn"])
+                noise_target = c.get(key, c["c_crossattn"])
                 gamma = cads_gamma(timestep)
-                for i in range(apply_to.size(dim=0)):
+                for i in range(noise_target.size(dim=0)):
                     if cond_or_uncond[i % len(cond_or_uncond)] == skip:
                         continue
-                    apply_to[i] = cads_noise(gamma, apply_to[i])
+                    noise_target[i] = cads_noise(gamma, noise_target[i])
 
             if previous_wrapper:
                 return previous_wrapper(apply_model, args)