diff --git a/__init__.py b/__init__.py
index 3cda6fd..1c82441 100644
--- a/__init__.py
+++ b/__init__.py
@@ -21,9 +21,9 @@ class CADS:
                 "noise_scale": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.25}),
                 "t1": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.6}),
                 "t2": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.9}),
-                "rescale": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.0}),
             },
             "optional": {
+                "rescale": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.0}),
                 "start_step": ("INT", {"min": -1, "max": 10000, "default": -1}),
                 "total_steps": ("INT", {"min": -1, "max": 10000, "default": -1}),
             },
@@ -34,7 +34,7 @@ class CADS:
 
     CATEGORY = "utils"
 
-    def do(self, model, noise_scale, t1, t2, rescale, start_step=0, total_steps=0):
+    def do(self, model, noise_scale, t1, t2, rescale=0.0, start_step=-1, total_steps=-1):
         previous_wrapper = model.model_options.get("model_function_wrapper")
 
         im = model.model.model_sampling