diff --git a/js/parameterDisplay.js b/js/parameterDisplay.js
index ae0e61f..7ca1901 100644
--- a/js/parameterDisplay.js
+++ b/js/parameterDisplay.js
@@ -47,9 +47,15 @@ with aspect ratio ${message.text[0]}: ${message.text[2]} x ${message.text[3]}`;
 
                 const start_at_float = parseFloat(message.text[5])
                 const base_percentage = Math.round(start_at_float * 100) + "%";
-                const refiner_percentage = Math.round((1 - start_at_float) * 100) + "%";
-                const step_message = `Total steps: ${message.text[4]}, refiner start at ${base_percentage},
-Base steps: ${message.text[6]} (${base_percentage}), Refiner steps: ${message.text[7]} (${refiner_percentage})`;
+                // const refiner_percentage = Math.round((1 - start_at_float) * 100) + "%";
+                let step_message;
+                if (start_at_float === 1) {
+                    step_message = `Total steps: ${message.text[4]},
+Refiner off`;
+                } else {
+                    step_message = `Total steps: ${message.text[4]},
+Refiner start at step: ${message.text[6]} (${base_percentage})`;
+                }
 
                 this.widgets.find(obj => obj.name === "steps_display").value = step_message;
                 this.widgets.find(obj => obj.name === "aspect_ratio_display").value = ar_message;
diff --git a/nodes.py b/nodes.py
index 20718ca..af41836 100644
--- a/nodes.py
+++ b/nodes.py
@@ -509,7 +509,6 @@ class SDParameterGenerator:
         "INT",
         "INT",
         "INT",
-        "INT",
         "FLOAT",
         comfy.samplers.KSampler.SAMPLERS,
         comfy.samplers.KSampler.SCHEDULERS,
@@ -528,8 +527,7 @@ class SDParameterGenerator:
         "VAE",
         "SEED",
         "STEPS",
-        "BASE_STEPS",
-        "REFINER_STEPS",
+        "REFINER_START_STEP",
         "CFG",
         "SAMPLER_NAME",
         "SCHEDULER",
@@ -603,7 +601,7 @@ class SDParameterGenerator:
             f"Sampler: {sampler_name},\n"
             f"Scheduler: {scheduler},\n"
             f"Size: {str(width)}x{str(height)},\n"
-            f"Batch size: {str(batch_size)},\n"
+            f"Batch size: {str(batch_size)}\n"
         )
 
         return {
@@ -626,7 +624,6 @@ class SDParameterGenerator:
                     seed,
                     steps,
                     base_steps,
-                    refiner_steps,
                     cfg,
                     sampler_name,
                     scheduler,