diff --git a/modules/ui.py b/modules/ui.py
index 3f2455c3..aa647b96 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -942,7 +942,7 @@ def create_ui():
         settings.text_settings.change(fn=update_image_cfg_scale_visibility, inputs=[], outputs=[image_cfg_scale])
         demo.load(fn=update_image_cfg_scale_visibility, inputs=[], outputs=[image_cfg_scale])
 
-        modelmerger_ui.setup_ui(dummy_component=dummy_component, sd_model_checkpoint_component=main_entry.sd_model_checkpoint)
+        modelmerger_ui.setup_ui(dummy_component=dummy_component, sd_model_checkpoint_component=main_entry.ui_checkpoint)
 
         main_entry.forge_main_entry()
 
diff --git a/modules/ui_settings.py b/modules/ui_settings.py
index 1d5c504c..bc9c8296 100644
--- a/modules/ui_settings.py
+++ b/modules/ui_settings.py
@@ -330,8 +330,8 @@ class UiSettings:
         button_set_checkpoint.click(
             fn=button_set_checkpoint_change,
             _js="function(v){ var res = desiredCheckpointName; desiredCheckpointName = ''; return [res || v, null]; }",
-            inputs=[main_entry.sd_model_checkpoint, self.dummy_component],
-            outputs=[main_entry.sd_model_checkpoint, self.text_settings],
+            inputs=[main_entry.ui_checkpoint, self.dummy_component],
+            outputs=[main_entry.ui_checkpoint, self.text_settings],
         )
 
         component_keys = [k for k in opts.data_labels.keys() if k in self.component_dict]