diff --git a/gui/elements/ml_gui_elements.py b/gui/elements/ml_gui_elements.py
index 9580335b8..ba5e41445 100644
--- a/gui/elements/ml_gui_elements.py
+++ b/gui/elements/ml_gui_elements.py
@@ -1262,7 +1262,7 @@ def set_frame_settings(self, frame_settings):
             if "collection_mode" in frame_settings:
                 collection_mode = frame_settings["collection_mode"]
                 if "auto" in collection_mode:
-                    if "feature map" in collection_mode:
+                    if "feature_based" in collection_mode:
                         self.auto_mode.setCurrentIndex(1)
                     else:
                         self.auto_mode.setCurrentIndex(0)