diff --git a/src/gui/core/worker_thread.py b/src/gui/core/worker_thread.py
index a46f78c..02ed34c 100644
--- a/src/gui/core/worker_thread.py
+++ b/src/gui/core/worker_thread.py
@@ -326,11 +326,15 @@ class WorkerThread(QThread):
             'step4': 'step5_process_csv',
             'step5': 'step6_extract_spectra',
             'step7': 'step7_calc_indices',
+            'step7_index': 'step7_calc_indices',
             'step8': 'step8_train_ml',
+            'step8_ml_train': 'step8_train_ml',
             'step8_non_empirical_modeling': 'step8_non_empirical_modeling',
             'step8_qaa': 'step8_qaa_inversion',
             'step9': 'step9_watercolor_inversion',
+            'step9_ml_predict': 'step9_predict_ml',
             'step10': 'step4_sampling',
+            'step10_map': 'step10_map',
             'step11_ml': 'step9_predict_ml',
             'step11': 'step11_non_empirical_prediction',
             'step14': 'step10_map'
diff --git a/src/gui/panels/step8_ml_train_panel.py b/src/gui/panels/step8_ml_train_panel.py
index 1a7a977..1675c3e 100644
--- a/src/gui/panels/step8_ml_train_panel.py
+++ b/src/gui/panels/step8_ml_train_panel.py
@@ -345,5 +345,5 @@ class Step8MlTrainPanel(QWidget):
 
         main_window = self.window()
         if hasattr(main_window, 'run_single_step'):
-            pipeline_config = {'step8': config}
-            main_window.run_single_step('step8', pipeline_config)
\ No newline at end of file
+            pipeline_config = {'step8_ml_train': config}
+            main_window.run_single_step('step8_ml_train', pipeline_config)
\ No newline at end of file