diff --git a/nodes.py b/nodes.py
index d798913..e04da2f 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1221,7 +1221,7 @@ class gradient_scaling_pre_cfg_node:
                 return conds_out
             else:
                 target_scales = maximum_scale * mask_as_weight * strength + torch.full_like(conds_out[1], cond_scale) * (1 - mask_as_weight * strength)
-                conds_out[1]  = make_new_uncond_at_scale(conds_out[0],conds_out[1],cond_scale,maximum_scale)
+                conds_out[1]  = make_new_uncond_at_scale(conds_out[0],conds_out[1],cond_scale,target_scales)
                 return conds_out
 
         m = model.clone()
@@ -1323,4 +1323,4 @@ class GradientRGBImage:
 
         mask_gradient = gradient.expand(1, height, width, 1).squeeze(-1)
 
-        return (rgb_image,mask_gradient,)
\ No newline at end of file
+        return (rgb_image,mask_gradient,)