diff --git a/frontend/components/config_loader.py b/frontend/components/config_loader.py
index a586a7e..89976cd 100644
--- a/frontend/components/config_loader.py
+++ b/frontend/components/config_loader.py
@@ -11,8 +11,7 @@ def get_default_config_loader(controller_name: str):
     existing_configs = [config["id"].split("_")[0] for config in all_configs]
     default_dict = {"id": generate_random_name(existing_configs)}
     default_config = st.session_state.get("default_config")
-    config_controller_name = st.session_state.get("controller_name", controller_name)
-    st.write(f"controller_name: {controller_name} | config_controller_name: {config_controller_name}")
+    config_controller_name = default_config.get("controller_name", controller_name)
     if default_config is None or controller_name != config_controller_name:
         st.session_state["default_config"] = default_dict
     with st.expander("Configurations", expanded=True):