diff --git a/kohya_gui/class_advanced_training.py b/kohya_gui/class_advanced_training.py
index 18fa997..4e51e3b 100644
--- a/kohya_gui/class_advanced_training.py
+++ b/kohya_gui/class_advanced_training.py
@@ -599,9 +599,9 @@ class AdvancedTraining:
             )
             self.log_tracker_config = gr.Dropdown(
                 label="Log tracker config",
-                choices=[self.config.get("log_tracker_config_dir", "")]
+                choices=[self.config.get("advanced.log_tracker_config_dir", "")]
                 + list_log_tracker_config_files(self.current_log_tracker_config_dir),
-                value=self.config.get("log_tracker_config_dir", ""),
+                value=self.config.get("advanced.log_tracker_config_dir", ""),
                 info="Path to tracker config file to use for logging",
                 interactive=True,
                 allow_custom_value=True,
@@ -610,7 +610,7 @@ class AdvancedTraining:
                 self.log_tracker_config,
                 lambda: None,
                 lambda: {
-                    "choices": [self.config.get("log_tracker_config_dir", "")]
+                    "choices": [self.config.get("advanced.log_tracker_config_dir", "")]
                     + list_log_tracker_config_files(self.current_log_tracker_config_dir)
                 },
                 "open_folder_small",
@@ -625,7 +625,7 @@ class AdvancedTraining:
             )
             self.log_tracker_config.change(
                 fn=lambda path: gr.Dropdown(
-                    choices=[self.config.get("log_tracker_config_dir", "")]
+                    choices=[self.config.get("advanced.log_tracker_config_dir", "")]
                     + list_log_tracker_config_files(path)
                 ),
                 inputs=self.log_tracker_config,
diff --git a/kohya_gui/class_folders.py b/kohya_gui/class_folders.py
index a0467fb..dbdaf69 100644
--- a/kohya_gui/class_folders.py
+++ b/kohya_gui/class_folders.py
@@ -26,13 +26,13 @@ class Folders:
 
         # Set default directories if not provided
         self.current_output_dir = self.config.get(
-            "output_dir", os.path.join(scriptdir, "outputs")
+            "folders.output_dir", os.path.join(scriptdir, "outputs")
         )
         self.current_logging_dir = self.config.get(
-            "logging_dir", os.path.join(scriptdir, "logs")
+            "folders.logging_dir", os.path.join(scriptdir, "logs")
         )
         self.current_reg_data_dir = self.config.get(
-            "reg_data_dir", os.path.join(scriptdir, "reg")
+            "folders.reg_data_dir", os.path.join(scriptdir, "reg")
         )
 
         # Create directories if they don't exist
diff --git a/kohya_gui/dreambooth_folder_creation_gui.py b/kohya_gui/dreambooth_folder_creation_gui.py
index 4f07fcb..4b55fb5 100644
--- a/kohya_gui/dreambooth_folder_creation_gui.py
+++ b/kohya_gui/dreambooth_folder_creation_gui.py
@@ -250,9 +250,9 @@ def gradio_dreambooth_folder_creation_tab(
             util_training_dir_output = gr.Dropdown(
                 label="Destination training directory (where formatted training and regularisation folders will be placed)",
                 interactive=True,
-                choices=[config.get(key="train_data_dir", default="")]
+                choices=[config.get(key="model.train_data_dir", default="")]
                 + list_train_output_dirs(current_train_output_dir),
-                value=config.get(key="train_data_dir", default=""),
+                value=config.get(key="model.train_data_dir", default=""),
                 allow_custom_value=True,
             )
             create_refresh_button(
@@ -272,7 +272,7 @@ def gradio_dreambooth_folder_creation_tab(
             )
             util_training_dir_output.change(
                 fn=lambda path: gr.Dropdown(
-                    choices=[config.get(key="train_data_dir", default="")] + list_train_output_dirs(path)
+                    choices=[config.get(key="model.train_data_dir", default="")] + list_train_output_dirs(path)
                 ),
                 inputs=util_training_dir_output,
                 outputs=util_training_dir_output,