diff --git a/py/nodes/checkpoint_loader.py b/py/nodes/checkpoint_loader.py
index cf0db731..8fdb77dc 100644
--- a/py/nodes/checkpoint_loader.py
+++ b/py/nodes/checkpoint_loader.py
@@ -37,7 +37,7 @@ class CheckpointLoaderLM:
                             "control_after_generate to pick a random model on "
                             "every run."
                         ),
-                        "control_after_generate": True,
+                        "control_after_generate": "fixed",
                     },
                 ),
                 "base_model": (
diff --git a/py/nodes/unet_loader.py b/py/nodes/unet_loader.py
index d0f11598..cffbb8e0 100644
--- a/py/nodes/unet_loader.py
+++ b/py/nodes/unet_loader.py
@@ -52,7 +52,7 @@ class UNETLoaderLM:
                             "control_after_generate to pick a random model on "
                             "every run."
                         ),
-                        "control_after_generate": True,
+                        "control_after_generate": "fixed",
                     },
                 ),
                 "weight_dtype": (